Postman
Berkeley, CA

AI Engineer, Intern

OnsitePosted yesterdayVisa Sponsorship

We tailor your resume to this role and apply for you in seconds.

Apply to AI Engineer, Intern at Postman

Job details

Location
Berkeley, CA
Work type
Onsite
Visa
Sponsorship available
Posted
yesterday
Apply on
boards.greenhouse.io

About this role

Postman is the world’s leading API platform, used by more than 45 million+ developers and 500,000 organizations. They are seeking an AI Engineer Intern to work alongside their AI team on large-scale AI systems, focusing on model development, productionization, and operational rigor.

What you'll do:

  • Partner with product managers, designers, and engineers to translate product requirements into scoped AI problem statements
  • Prepare data pipeline and AI store design under senior engineer guidance
  • Build and validate AI models using PyTorch or JAX(Optax/Orbax / TensorStore/Grain) and similar tools
  • Run large-scale experiments, evaluate models against defined metrics, and support ablation studies and error analysis
  • Help build inference APIs and batch scoring workflows; integrate AI outputs with backend services
  • Support AIOps practices already in place on the team: model versioning, CI/CD pipelines, monitoring dashboards, and logging
  • Assist with model optimization work (quantization, batching, distillation) under supervision -this is a learning objective, not an expected independent deliverable
  • Contribute to root-cause analysis on production issues alongside senior engineers
  • Document experiments, design decisions, and runbooks so your work is legible to the next person who touches it
  • Flag fairness, interpretability, or privacy concerns you observe in data or model behavior….you're not expected to resolve these alone, but surfacing them is part of the job

What they're looking for:

  • Currently pursuing a BS/MS/PhD in Computer Science, Data Science, or a related field
  • Some hands-on experience with ML systems - course projects, research, hackathons, or a prior internship all count
  • Solid Python/Rust fundamentals: data structures, functions, basic testing, comfort reading and writing code reviews
  • Exposure to at least one AI/ML frameworks (PyTorch or JAX) and basic data tooling (Pandas, SQL)
  • Clear written and verbal communication
  • Familiarity with a cloud platform (AWS/GCP/Azure)
  • Docker
  • A model-serving tool (vLLM, Triton, Ray Serve)

Benefits:

  • In addition to Postman's pay-on-performance philosophy
  • Flexible schedule working with a fun, collaborative team
  • Full medical coverage
  • Flexible PTO
  • Wellness reimbursement
  • A monthly lunch stipend
  • Wellness programs
  • Frequent and fascinating team-building events
  • Donation-matching program
  • In office 5 days a week for all roles based out of our hubs in San Francisco Bay Area, Boston, Austin, New York City, Tokyo and London
Ready to apply to Postman?
We tailor your resume to this role and apply for you.

About Postman

Postman
Berkeley, CA