ML Research Engineer: Large-Scale Training & Tools

Mistral.ai; Paris; On-site; EUR 70,000 - 120,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Relocation support

Wellness programs

Job summary

Mistral.ai in Paris, France is seeking a Research Engineer – ML track to build and optimise the large-scale learning systems powering open-weight models. You’ll work with Research Scientists and may join Platform RE Team or Embedded RE Team to turn ideas into scalable code.

You will accelerate researchers, interface research with production, and conduct experiments on the latest DL techniques, writing Python and delivering production-grade prototypes.

Qualifications

• Master’s or PhD in Computer Science or equivalent.
• 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; CUDA or data‑pipeline chops a plus.
• 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.

Skills

PyTorch

JAX

TensorFlow

Distributed training

Deep learning

NLP/LLMs

CUDA

Software design

CI/CD

Collaboration

Education

Master’s or PhD in Computer Science

Tools

DeepSpeed

FSDP

SLURM

Kubernetes

Job description

Mistral.ai in Paris, France is seeking a Research Engineer – ML track to build and optimise the large-scale learning systems powering open-weight models. You’ll work with Research Scientists and may join Platform RE Team or Embedded RE Team to turn ideas into scalable code.

You will accelerate researchers, interface research with production, and conduct experiments on the latest DL techniques, writing Python and delivering production-grade prototypes.