Senior AI/ML Engineer

About the role

Our client is a well‑funded AI startup building production‑grade ML infrastructure for enterprise customers. They seek a Senior AI/ML Engineer to own model training pipelines, evaluation systems, and inference serving at scale. The role is full‑time and on‑site in San Francisco.

Key responsibilities

• Design and ship end‑to‑end ML systems, including data pipelines, training, evaluation, and deployment.
• Own model performance, latency, and cost trade‑offs in production.
• Build evaluation harnesses and offline benchmarks for fast iteration.
• Collaborate with product teams to translate ambiguous goals into measurable model improvements.
• Mentor other engineers on ML best practices and code quality.

Required profile

• 4+ years of applied ML engineering in production environments.
• Hands‑on experience with LLMs, fine‑tuning, RAG, or large‑scale recommender systems.
• Strong Python and PyTorch (or JAX) fundamentals.
• Experience with distributed training, GPU optimization, or inference serving.
• Pragmatic about trade‑offs between research‑grade and ship‑grade work.

Required skills

• Python
• PyTorch
• JAX
• Distributed training
• GPU optimization
• Inference serving
• Large Language Models (LLMs)
• Fine‑tuning
• Retrieval‑augmented generation (RAG)
• Recommender systems