Research Engineer, ML Systems & Production Tools

Mistral; Warszawa; On-site; PLN 230,000 - 380,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Retirement plans

Relocation support

Wellness programs

Meal allowances

Transportation allowances

Job summary

Mistral is recruiting a Research Engineer – ML track to build and optimize large-scale learning systems powering open-weight models. You will collaborate with Research Scientists and move between research and production within a distributed team across Europe, North America, Asia, and the Middle East.

In this role you’ll implement and benchmark ML algorithms, design robust tools, and deliver prototypes that scale to production.

Qualifications

• Master’s or PhD in Computer Science (or equivalent proven track record).
• 4+ years working on large-scale ML codebases.
• Experience with PyTorch, JAX or TensorFlow and distributed training (DeepSpeed / FSDP / SLURM / K8s).
• Experience in deep learning, NLP or LLMs; CUDA or data-pipeline chops is a bonus.

Responsibilities

• Accelerate researchers by handling heavy parts of large-scale ML pipelines and building robust tools.
• Interface research with production: integrate checkpoints, streamline evaluation, and expose APIs.
• Conduct experiments on latest deep-learning techniques and benchmarks.
• Design, implement and benchmark ML algorithms; write clean Python code.
• Deliver prototypes that become production-grade components for Le Chat and our enterprise API.

Skills

PyTorch

JAX

TensorFlow

DeepSpeed / FSDP

SLURM / Kubernetes

NLP / LLMs

Education

Master’s or PhD in Computer Science

Tools

CUDA

CI/CD

Job description

Mistral is recruiting a Research Engineer – ML track to build and optimize large-scale learning systems powering open-weight models. You will collaborate with Research Scientists and move between research and production within a distributed team across Europe, North America, Asia, and the Middle East.

In this role you’ll implement and benchmark ML algorithms, design robust tools, and deliver prototypes that scale to production.