Senior ML Engineer — Scale Training Infra & AI Deployments

Best AI Tools Wiki; San Francisco, Northern (CA, KY); Hybrid; USD 180,000 - 350,000; Full time

Benefits offered by this job

Equity package

Health insurance

Unlimited PTO

Learning budget

Free meals

401(k) matching

Job summary

OpenAI in San Francisco is seeking an experienced ML Engineer to help build and optimize large-scale ML systems powering next-generation AI models. You will contribute to training infrastructure, model optimization, and deployment pipelines for flagship products including the API platform and ChatGPT.

Qualifications include 5+ years in ML engineering, strong Python and PyTorch or JAX, and an advanced degree; experience with GPU/TPU clusters.

Qualifications

• 5+ years of experience in machine learning engineering.
• Strong proficiency in Python and PyTorch or JAX.
• MS or PhD in Computer Science, Machine Learning, or related field.
• Experience with GPU/TPU cluster management.

Responsibilities

• Build training infrastructure for large-scale models.
• Optimize models for performance and efficiency.
• Develop and maintain deployment pipelines for API and products.

Skills

ML Engineering

Python

PyTorch

JAX

Distributed Training

CUDA

Transformers

RLHF

Education

MS or PhD in Computer Science, Machine Learning, or related field

Tools

GPU/TPU cluster management

Job description

OpenAI in San Francisco is seeking an experienced ML Engineer to help build and optimize large-scale ML systems powering next-generation AI models. You will contribute to training infrastructure, model optimization, and deployment pipelines for flagship products including the API platform and ChatGPT.

Qualifications include 5+ years in ML engineering, strong Python and PyTorch or JAX, and an advanced degree; experience with GPU/TPU clusters.