Research Engineer, Machine Learning

Mistral AI•Palo Alto, CA

About The Position

About the Research Engineering team The team spans Platform (shared infra & clean code) and Embedded (inside research squads). Engineers can move along the research↔production spectrum as needs or interests evolve. As a Research Engineer – ML track, you’ll build and optimise the large-scale learning systems that power our open-weight models. Working hand-in-hand with Research Scientists, you’ll either join: - Platform RE Team: Enhance the shared training framework, data pipelines and cluster tooling used by every team; or - Embedded RE Team: Sit inside a research squad (Alignment, Pre-training, Multimodal, …) and turn fresh ideas into repeatable, scalable code.

Requirements

Master’s or PhD in Computer Science (or equivalent proven track record).
4 + years working on large-scale ML codebases.
Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed / FSDP / SLURM / K8s).
Experience in deep learning, NLP or LLMs; bonus for CUDA or data-pipeline chops.
Strong software-design instincts: testing, code review, CI/CD.
Self-starter, low-ego, collaborative.

Responsibilities

Accelerate researchers by taking on the heavy parts of large-scale ML pipelines and building robust tools.
Interface cutting-edge research with production: integrate checkpoints, streamline evaluation, and expose APIs.
Conduct experiments on the latest deep-learning techniques (sparsified 70 B + runs, distributed training on thousands of GPUs).
Design, implement and benchmark ML algorithms; write clear, efficient code in Python .
Deliver prototypes that become production-grade components for Le Chat and our enterprise API.

Benefits

Competitive salary and equity.
Medical/Dental/Vision covered for you and your family.
401K (6% matching)
18 days PTO
Reimburse office parking charges, or $120/month for public transport
$120/month reimbursement for gym membership
$400 monthly allowance for meals
Visa sponsorship
BetterUp coaching on a voluntary basis
20 weeks of paid leave for all primary parents.
Reserved daycare seats or financial assistance for working parents.
Financial assistance for moving expenses, visa sponsorship, and settling-in services for you and your family.
Monthly stipends, vouchers, or catered in-office meals, depending on your office location.
Monthly allowances for public transport, parking, or sustainable mobility
Subsidies or memberships for gyms, wellness programs, or fitness activities