AI Engineer, Intern New Berkeley, California, United States

Postman, Inc.; San Francisco, Northern (CA, KY); On-site; USD 38,572 - 55,104; Full time

Job summary

Postman, Inc. in Berkeley is seeking an AI Engineer Intern to join our AI team. You will own discrete pieces of real systems under mentorship, from data pipeline design to production deployment.

You will collaborate with product managers, designers, and engineers, build AI models with PyTorch or JAX, run large-scale experiments, and help advance the API-first platform.

Responsibilities

• 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 requirements.

Skills

AI model dev

Experimentation

Team collaboration

Tools

PyTorch

JAX

Optax

Orbax

TensorStore

Grain

Job description

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.

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

• 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