AI Engineer - Intern
About Us
Postman is the world’s leading API platform, used by more than 45 million developers and 500,000 organizations, including 98% of the Fortune 500. We help developers and professionals across the globe build the API-first world by simplifying each step of the API lifecycle and streamlining collaboration—enabling users to create better APIs, faster.
About the Role
We're seeking an AI Engineer Intern to work alongside our AI team on large-scale AI and Agentic systems from data pipeline to production deployment. This role is scoped for someone with foundational experience who wants to deepen it: you'll own discrete pieces of real systems under the mentorship of senior engineers, not shadow work or isolated coursework-style projects.
Responsibilities
- Model Development
- 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.
- Productionization
- 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.
- Operational Rigor
- 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.
Requirements
- 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 framework (PyTorch or JAX) and basic data tooling (Pandas, SQL).
- Clear written and verbal communication. You'll be documenting your own work and asking questions across product, design, and engineering.
Nice to Have
- Familiarity with a cloud platform (AWS/GCP/Azure), Docker, or a model-serving tool (vLLM, Triton, Ray Serve).
Benefits
- Full medical coverage.
- Flexible PTO.
- Wellness reimbursement.
- Monthly lunch stipend.
- Wellness programs for physical and mental health.
- Frequent team-building events.
- Donation-matching program.
Schedule
This role is in-office 5 days a week for all hub locations (San Francisco Bay Area, Boston, Austin, New York City, Tokyo, and London). For Bangalore-based roles, employees currently work in the office three days a week and will transition to five days per week by the end of the year.