AI Engineer, Intern

Postman; Berkeley (CA); On-site; USD 28,000 - 48,000; Full time

Benefits offered by this job

Full medical coverage

Flexible PTO

Wellness reimbursement

Monthly lunch stipend

Donation matching program

Job summary

Postman is seeking an AI Engineer Intern to work with the AI team on large-scale AI and Agentic systems from data pipeline to production. You will own discrete pieces of real systems under mentorship, contributing to model development and productionization with guidance from senior engineers.

Ideal candidates are pursuing BS/MS/PhD in CS/Data Science, have some ML project experience, and are comfortable with Python and at least one ML framework.

Qualifications

• 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).
• Nice to have, not required: familiarity with a cloud platform (AWS/GCP/Azure), Docker, or a model-serving tool (vLLM, Triton, Ray Serve).
• Clear written and verbal communication. You'll documenting your own work and asking questions across product, design, and engineering.

Responsibilities

• Translate product requirements into scoped AI problem statements with product teams.
• Prepare data pipeline and AI store design under senior engineer guidance.
• Build and validate AI models using PyTorch or JAX and similar tools.
• Run large-scale experiments, evaluate models against metrics, and support ablation studies.

Skills

Communication

ML systems exposure

Problem solving

Education

BS/MS/PhD in CS/Data Science or related

Tools

Python

Rust

PyTorch

JAX

Job description

Who Are We?

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. Postman is helping 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.

The company is headquartered in San Francisco and has offices in Boston, New York, Austin, Tokyo, London, and Bangalore - where Postman was founded. Postman is privately held, with funding from Battery Ventures, BOND, Coatue, CRV, Insight Partners, and Nexus Venture Partners. Learn more at postman.com or connect with Postman on X via @getpostman.

P.S: We highly recommend reading the "API-First World" graphic novel to understand the bigger picture and our vision at Postman.

The Opportunity

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.

What You'll Do

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.

About You

• 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. We are not requiring 1+ years of production experience for this role; that bar belongs on a full-time posting, not an internship.
• 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).
• Nice to have, not required: familiarity with a cloud platform (AWS/GCP/Azure), Docker, or a model-serving tool (vLLM, Triton, Ray Serve).
• Clear written and verbal communication. You'll documenting your own work and asking questions across product, design, and engineering.

What Else?

In addition to Postman's pay-on-performance philosophy, and a flexible schedule working with a fun, collaborative team, Postman offers a comprehensive set of benefits, including full medical coverage, flexible PTO, wellness reimbursement, and a monthly lunch stipend. Along with that, our wellness programs will help you stay in the best of your physical and mental health. Our frequent and fascinating team-building events will keep you connected, while our donation-matching program can support the causes you care about. We're building a long-term company with an inclusive culture where everyone can be the best version of themselves.

At Postman we value in person collaboration. We are 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. For roles based in Bangalore, employees currently work in the office three days a week and will transition to five days per week by the end of the year. We were thoughtful in our approach which is based on collaboration and grounded in feedback from our workforce, leadership team, and peers. The benefits of our in office model will be shared knowledge, brainstorming sessions, communication, and building trust in-person that cannot be replicated via zoom.

Our Values

At Postman, we create with the same curiosity that we see in our users. We value transparency and honest communication about not only successes, but also failures. In our work, we focus on specific goals that add up to a larger vision. Our inclusive work culture ensures that everyone is valued equally as important pieces of our final product. We are dedicated to delivering the best products we can.

Equal opportunity

Postman is an Equal Employment Opportunity and Affinite Action Employer. Qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender perception or identity, national origin, age, marital status, protected veteran status, or disability status. Headhunters and recruitment agencies may not submit resumes/CVs through this website or directly to managers. Postman does not accept unsolicited headhunter and agency resumes. Postman will not pay fees to any third-party agency or company that does not have a signed agreement with Postman.