Research Engineer, Machine Learning

11 Months ago • 4 Years + • Research Development

Job Summary

Job Description

Mistral AI is seeking a Research Engineer specializing in Machine Learning to build and optimize large-scale learning systems for their open-weight models. Working closely with Research Scientists, you will contribute to either the Platform RE Team, enhancing shared training frameworks, data pipelines, and cluster tooling, or the Embedded RE Team, integrating novel ideas into repeatable, scalable code within research squads. Responsibilities include accelerating researchers by managing large-scale ML pipelines, interfacing research with production by integrating checkpoints and streamlining evaluations, conducting experiments on cutting-edge deep learning techniques, designing and implementing ML algorithms, and delivering production-grade prototypes for products like 'Le Chat' and enterprise APIs. The role requires strong software design instincts, proficiency in Python, and experience with distributed training frameworks.
Must have:
  • Master's or PhD in Computer Science or equivalent
  • 4+ years on large-scale ML codebases
  • Hands-on with PyTorch, JAX, or TensorFlow
  • Comfortable with distributed training
  • Experience in deep learning, NLP, or LLMs
  • Strong software-design instincts
  • Self-starter, low-ego, collaborative
Good to have:
  • CUDA experience
  • Data-pipeline experience
Perks:
  • Competitive cash salary and equity
  • Daily lunch vouchers (France)
  • Monthly contribution to Gympass subscription (France)
  • Monthly contribution to a mobility pass (France)
  • Full health insurance for you and your family (France)
  • Generous parental leave policy (France)
  • Visa sponsorship (France)
  • Insurance (UK)
  • Reimbursement for office parking or public transport (UK)
  • Reimbursement for gym membership (UK)
  • Monthly meal allowance (UK)
  • Pension plan (UK)

Job Details

About Mistral

At Mistral AI, we believe in the power of AI to simplify tasks, save time, and enhance learning and creativity. Our technology is designed to integrate seamlessly into daily working life.

We democratize AI through high-performance, optimised, open-source and cutting-edge models, products and solutions. Our comprehensive AI platform is designed to meet the needs of enterprises, whether on-premises or in cloud environments. Our offerings include Le Chat, the AI assistant for life and work.

We are a dynamic, collaborative team passionate about AI and its potential to transform society. Our diverse workforce thrives in competitive environments and is committed to driving innovation. Our teams are distributed between France, the USA, the UK, Germany and Singapore. We are creative, low-ego and team-spirited. Join us to be part of a pioneering company shaping the future of AI. Together, we can make a meaningful impact. See more about our culture on https://mistral.ai/careers.

Role Summary 

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.

What will you do

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

About you

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

Similar Jobs

Jash entertainment  - C++ Developer

Jash entertainment

Ahmedabad, Gujarat, India (On-Site)
3 Months ago
Roblox - Senior / Principal Model Optimization Engineer

Roblox

San Mateo, California, United States (On-Site)
1 Month ago
NVIDIA - Deep Learning Engineer, Datacenters

NVIDIA

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
NVIDIA - Senior Solutions Architect, Omniverse Platform

NVIDIA

Beijing, Beijing, China (On-Site)
4 Months ago
CharacterAI - Research Engineer, ML Systems

CharacterAI

New York, New York, United States (On-Site)
4 Months ago
Sully.ai - Senior ML/NLP Engineer (Contract)

Sully.ai

United States (Remote)
3 Weeks ago
Reddit - Senior Software Engineer, AI Enablement

Reddit

(Remote)
2 Months ago
bytedance - Machine Learning Scientist, Scaling AI for Biology

bytedance

Seattle, Washington, United States (On-Site)
9 Months ago
Aledade - Technical Program Manager (Data & AI)

Aledade

United States (Remote)
2 Months ago
Highspot - Senior Machine Learning Engineer

Highspot

Seattle, Washington, United States (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Ansys - R&D Engineer II

Ansys

Waterloo, Ontario, Canada (On-Site)
1 Month ago
Perplexity - Tech Lead - Acceleration

Perplexity

New York, United States (On-Site)
1 Month ago
Shield AI - Staff Engineer, Reinforcement Learning (R3639)

Shield AI

San Diego, California, United States (On-Site)
2 Weeks ago
NVIDIA - Deep Learning Engineer, Datacenters

NVIDIA

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
Roblox - Senior / Principal Model Optimization Engineer

Roblox

San Mateo, California, United States (On-Site)
1 Month ago
bytedance - Tech Lead - Architect / Researcher - DPU

bytedance

Seattle, Washington, United States (On-Site)
5 Months ago
bytedance - AI Model Optimization Engineer

bytedance

San Jose, California, United States (On-Site)
4 Months ago
bytedance - Software Engineer, Machine Learning Training

bytedance

Singapore (On-Site)
4 Months ago
bytedance - Tech Lead, Research Scientist, Cloud & AI Computing - DPU/GPU/CPU

bytedance

Seattle, Washington, United States (On-Site)
4 Months ago
Intangible - Applied AI Engineer (Image/Video Diffusion)

Intangible

United States (Remote)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Paris, Île-de-France, France

Ubisoft - Principal R&D Scientist on Bots & Behaviors

Ubisoft

Bordeaux, Nouvelle-Aquitaine, France (Hybrid)
6 Months ago
Condé Nast - Shopping / Stylist Assistant Vogue Intern

Condé Nast

Paris, Île-de-France, France (On-Site)
4 Weeks ago
Thales - Expert Technical Support Cybersecurity

Thales

Vélizy-Villacoublay, Île-de-France, France (Hybrid)
4 Weeks ago
Sorare - Job not found

Sorare

Paris, Île-de-France, France (On-Site)
3 Weeks ago
Luni - iOS Developer

Luni

Bordeaux, Nouvelle-Aquitaine, France (Hybrid)
3 Years ago
Ubisoft - Accounts Receivable and Collections Assistant - Internship

Ubisoft

Paris, Île-de-France, France (On-Site)
4 Months ago
PwC - Management Group Audit financier

PwC

Nantes, Pays De La Loire, France (On-Site)
2 Months ago
Slite - Account Executive

Slite

France (Remote)
1 Month ago
Square - Accounting Collaborator M/F

Square

Grenoble, Auvergne-Rhône-Alpes, France (Hybrid)
2 Weeks ago
Luni - Head of Legal

Luni

Bordeaux, Nouvelle-Aquitaine, France (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Lambda - Applied ML Engineer - Inference

Lambda

San Francisco, California, United States (Hybrid)
2 Months ago
Unity - Senior Machine Learning/MLOps Developer

Unity

Montreal, Quebec, Canada (On-Site)
10 Months ago
Canva - Senior Machine Learning Engineer - Content Management & Distribution

Canva

Brisbane, Queensland, Australia (Remote)
1 Month ago
Apple - ML Framework (MetalLM) Engineer

Apple

Cupertino, California, United States (On-Site)
2 Months ago
Zinnia - ML Ops Engineer

Zinnia

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
CharacterAI - Research Engineer, Post-Training

CharacterAI

United States (On-Site)
1 Year ago
Meta - Research Scientist Intern, Smart Glasses in Wearables AI (PhD)

Meta

Menlo Park, California, United States (On-Site)
9 Months ago
HCL Tech - AI Program Management

HCL Tech

Brazil (Remote)
2 Months ago
Xsolla - Machine Learning Engineer

Xsolla

Montreal, Quebec, Canada (Remote)
3 Months ago
Reddit - Senior Machine Learning Engineer

Reddit

United States (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Paris, Île-de-France, France (Hybrid)

Paris, Île-de-France, France (On-Site)

New York, United States (Hybrid)

Palo Alto, California, United States (On-Site)

Paris, Île-de-France, France (On-Site)

Paris, Île-de-France, France (Hybrid)

Palo Alto, California, United States (Hybrid)

Paris, Île-de-France, France (Hybrid)

Paris, Île-de-France, France (On-Site)

Paris, Île-de-France, France (Hybrid)

View All Jobs

Get notified when new jobs are added by Mistral AI

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug