Research Engineer, Machine Learning

Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. 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 join either the Platform RE Team or the Embedded RE Team.

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

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; Strong software-design instincts: testing, code review, CI/CD; Self-starter, low-ego, collaborative

Nice to have

CUDA or data-pipeline chops

Benefits

Healthcare coverage; Parental leave; Retirement plans; Relocation support; Wellness programs; Meal and transportation allowances; Location-specific perks