Mistral AI · Paris/London

Research Engineer, Machine Learning

29.8.2024

Description


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.

Location: Paris / London (hybrid) or remote EU/UK with one hub day per month.


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

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.

Benefits


France
💰 Competitive cash salary and equity
🥕 Food: Daily lunch vouchers
🥎 Sport: Monthly contribution to a Gympass subscription
🚴 Transportation: Monthly contribution to a mobility pass
🧑‍⚕️ Health: Full health insurance for you and your family
🍼 Parental: Generous parental leave policy
🌎 Visa sponsorship

UK
💰 Competitive cash salary and equity
🚑 Insurance
🚴 Transportation: Reimburse office parking charges, or  £90 per month for public transport
🥎 Sport:  £90 per month reimbursement for gym membership
🥕 Meal voucher: £200 monthly allowance for meals
💰 Pension plan: SmartPension (percentages are 5% Employee & 3% Employer)

Application

View listing at origin and apply!