Open-Source Software, Machine Learning Engineer
Remote • Paris • Full-time
Posted 1y ago
Job Location
Paris
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, products, and solutions. We are a dynamic, collaborative team passionate about AI's potential to transform society, with teams distributed globally. We are seeking an Open-Source Software, Machine Learning Engineer to join our OSS team, which is embedded within our Science team. This role is critical in helping turn research breakthroughs into tangible solutions and improving Mistral's open-source ecosystem by open-sourcing state-of-the-art models and maintaining our publicly available libraries.
Responsibilities
- Release models to open-source platforms and libraries (e.g., vLLM, GitHub, Hugging Face).
- Maintain Mistral's open-source libraries (mistral-common, mistral-finetune, mistral-inference).
- Create and maintain internal and external facing tooling and services.
- Implement and optimize open-source and internal libraries for performance and accuracy, ensuring production readiness.
- Collaborate with the open-source community (PyTorch, vLLM, Hugging Face).
Requirements
- Master's degree in Computer Science, Machine Learning, Data Science, or a related field.
- Experience contributing to popular open-source libraries (e.g., PyTorch, Tensorflow, JAX, vLLM, Transformers, Llama.cpp).
- Expert programming skills in Python and PyTorch.
- Proficiency in MLOps.
- Deep understanding of machine learning approaches, especially LLMs and algorithms.
- Experience with training and fine-tuning large language models (e.g., distillation, supervised fine-tuning, policy optimization).
- Experience working with Slurm.
- Experience working with research teams.
- Experience as a core-maintainer of a popular ML open-source library.