Research Engineer, ML - Scale Large-Scale DL Pipelines

Mistral; Paris; Sur place; EUR 90 000 - 150 000; Plein temps

Avantages offerts par ce poste

Salary + equity

Lunch vouchers

Gympass

Mobility pass

Health insurance

Parental leave

Visa sponsorship

Résumé du poste

Mistral is seeking a Research Engineer – ML track to join the Research Engineering team, spanning Platform and Embedded squads. You will build and optimize large-scale learning systems that power our open-weight models, collaborating with Research Scientists to turn ideas into repeatable, scalable code.

You will work on training frameworks, data pipelines and distributed training, crafting production-grade components and APIs for Le Chat and enterprise solutions.

Qualifications

• Master’s or PhD in Computer Science or equivalent with a proven track record.
• Experience with PyTorch, JAX or TensorFlow and distributed training.
• Strong software design, testing, code review, and CI/CD practices.

Responsabilités

• Accelerate researchers by handling heavy parts of large-scale ML pipelines and building robust tools.
• Interface cutting-edge research with production: integrate checkpoints, streamline evaluation, and expose APIs.
• 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

Python

PyTorch

JAX/TensorFlow

Distributed training

CI/CD

Team collaboration

Formation

Master's or PhD in Computer Science

Outils

DeepSpeed

FSDP

SLURM

Kubernetes (K8s)

CUDA

Description du poste

Mistral is seeking a Research Engineer – ML track to join the Research Engineering team, spanning Platform and Embedded squads. You will build and optimize large-scale learning systems that power our open-weight models, collaborating with Research Scientists to turn ideas into repeatable, scalable code.

You will work on training frameworks, data pipelines and distributed training, crafting production-grade components and APIs for Le Chat and enterprise solutions.