ML Research Engineer — Large-Scale Training & Tools

The Resume Database; Palo Alto (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

A leading AI-focused technology company in California is seeking an experienced Research Engineer to build and optimize large-scale learning systems for open-weight models. Responsibilities include integrating research with production, conducting experiments on cutting-edge ML techniques, and delivering prototypes that evolve into production-grade components. Candidates should have a Master's or PhD in Computer Science and over 4 years of experience with ML codebases and relevant tools. This role offers competitive compensation in a dynamic work environment.

Qualifications

• 4+ years of experience in developing large-scale ML systems.
• Experience with distributed training and deep learning techniques.
• Strong software-design instincts, including code review and CI/CD practices.

Responsibilities

• Build and optimize large-scale learning systems for open-weight models.
• Integrate research with production, streamlining evaluation and exposing APIs.
• Conduct experiments with advanced deep-learning methods.
• Deliver prototypes that evolve into production-grade components.

Skills

Large-scale ML codebases

PyTorch

JAX

TensorFlow

DeepSpeed

FSDP

SLURM

K8s

Deep learning

NLP

Education

Master's or PhD in Computer Science

Job description

A leading AI-focused technology company in California is seeking an experienced Research Engineer to build and optimize large-scale learning systems for open-weight models. Responsibilities include integrating research with production, conducting experiments on cutting-edge ML techniques, and delivering prototypes that evolve into production-grade components. Candidates should have a Master's or PhD in Computer Science and over 4 years of experience with ML codebases and relevant tools. This role offers competitive compensation in a dynamic work environment.