Research Engineer, Machine Learning

Mistral AI

$130K — $180K *
Enterprise Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Master's or PhD in Computer Science or equivalent experience
  • 4+ years of experience with large-scale ML codebases
  • Proficient in PyTorch, JAX, or TensorFlow
  • Experience in deep learning, NLP, or LLMs
  • Familiarity with distributed training tools like DeepSpeed or SLURM
  • Strong skills in software design practices, including testing and CI/CD
  • Collaborative team player with a self-starter attitude

Responsibilities

  • Accelerate researchers by handling large-scale ML pipeline tasks
  • Integrate cutting-edge research with production systems
  • Conduct experiments on advanced deep-learning techniques
  • Design and implement ML algorithms in Python
  • Benchmark and prototype production-grade components

Benefits

  • Healthcare coverage
  • Parental leave
  • Retirement plans
  • Relocation support
  • Wellness programs
  • Meal and transportation allowances
  • Location-specific perks
Full Job Description
The Role

The team spans Platform (shared infra & clean code) and Embedded (inside research squads). Engineers can move along the research14production spectrum as needs or interests evolve.

As a Research Engineer - ML track, youll build and optimise the large-scale learning systems that power our open-weight models. Working hand-in-hand with Research Scientists, youll 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.

What You Will 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.

What Were Looking For
 Masters 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.

What We Offer

We offer a comprehensive benefits package designed to support your well-being, growth, and work-life balance. Benefits vary by country and may include healthcare coverage, parental leave, retirement plans, relocation support, wellness programs, meal and transportation allowances, and other location-specific perks.

For the most up-to-date details on benefits available in your location, please refer to our Benefits page.

Similar Jobs

More Jobs at Mistral AI

More Enterprise Technology Jobs

Find similar Research Engineer, Machine Learning jobs: