Research Engineer, Machine Learning
Remote • Palo Alto • Full-time
Posted 6mo ago
Job Location
Palo Alto
Tech Stack
Remote Work Policy
Fully remote
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
Mistral AI is democratizing AI through high-performance, optimized, open-source models and solutions. We are a dynamic, collaborative team passionate about AI's potential to transform society, with a diverse workforce driving innovation. As a Research Engineer – ML track, you will build and optimize large-scale learning systems powering our open-weight models. You will work hand-in-hand with Research Scientists, either enhancing the shared training framework and data pipelines or embedding within a research squad to turn fresh ideas into scalable code.
Responsibilities
- Accelerate researchers by taking on heavy parts of large-scale ML pipelines and building robust tools.
- Interface cutting-edge research with production by integrating checkpoints, streamlining evaluation, and exposing APIs.
- Conduct experiments on the latest deep-learning techniques, including large-scale distributed training.
- Design, implement, and benchmark ML algorithms, writing clear, efficient Python code.
- Deliver prototypes that become production-grade components for products like Le Chat and enterprise APIs.
Requirements
- Master's or PhD in Computer Science or equivalent proven track record.
- 4+ years of experience working on large-scale ML codebases.
- Hands-on experience with PyTorch, JAX, or TensorFlow.
- Comfortable with distributed training frameworks (DeepSpeed, FSDP, SLURM, K8s).
- Experience in deep learning, NLP, or LLMs.
- Strong software design instincts, including testing, code review, and CI/CD.
- Self-starter, low-ego, and collaborative attitude.
Benefits
- Competitive salary and equity.
- Medical, Dental, and Vision healthcare coverage for you and your family.
- 401K with 6% matching.
- 18 days of Paid Time Off (PTO).
- Reimbursement for office parking charges or $120/month for public transport.
- $120/month reimbursement for gym membership.
- $400 monthly allowance for meals.
- Visa sponsorship.
- Voluntary BetterUp coaching.