Research Engineer, ML Systems for Scalable AI

Mistral.ai; Paris; Sur place; EUR 90 000 - 130 000; Plein temps

Résumé du poste

Mistral is seeking a Research Engineer – ML track to build and optimise large-scale learning systems powering open-weight models. You will work with Research Scientists and join either Platform RE Team or Embedded RE Team to turn ideas into scalable code.

You will design and implement ML algorithms, run distributed training, and deliver production-ready prototypes for their enterprise API while collaborating across Europe, North America, Asia and the Middle East.

Qualifications

• 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.

Responsabilités

• Accelerate researchers by taking on 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.

Connaissances

PyTorch

JAX

TensorFlow

Distributed training

CUDA

Data pipelines

CI/CD

Software design

Testing

Code review

Formation

Master’s or PhD in Computer Science

Outils

DeepSpeed

FSDP

SLURM

Kubernetes (K8s)

Description du poste

Mistral is seeking a Research Engineer – ML track to build and optimise large-scale learning systems powering open-weight models. You will work with Research Scientists and join either Platform RE Team or Embedded RE Team to turn ideas into scalable code.

You will design and implement ML algorithms, run distributed training, and deliver production-ready prototypes for their enterprise API while collaborating across Europe, North America, Asia and the Middle East.