Senior Machine Learning Engineer

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

Join OpenAI to build and optimize large-scale machine learning systems powering the next generation of AI models. You will work on training infrastructure, model optimization, and deployment pipelines for our flagship products including ChatGPT and the API platform.

Requirements

• 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

Nice to Have

• Experience with transformer architectures
• Publications in top ML conferences (NeurIPS, ICML, ICLR)
• Experience with RLHF or preference learning
• Contributions to open-source ML projects

Competitive equity package

Comprehensive health, dental, and vision insurance

Unlimited PTO

$5,000 annual learning budget

Free lunch and dinner

401(k) with company match

Skills

Python PyTorch Distributed Training CUDA Transformers RLHF

Vincony has all 400+ AI models in one place — compare responses, AI debate, Image/Video/Voice generator, and 20 more tools to help you learn and build with AI.