Machine Learning Engineer, Open-Source Software

Machine Learning Engineer, Open-Source Software - Paris/London Join to apply for the Machine Learning Engineer, Open-Source Software - Paris/London role at Mistral AI 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, optimized, open-source and cutting-edge models, products and solutions. Our comprehensive AI platform is designed to meet enterprise needs, 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, USA, 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 You will be in charge of open-sourcing state-of-the-art models, whilst maintaining and improving Mistral’s publicly available libraries. Your work is critical in helping turn research breakthroughs into tangible solutions and improve Mistral's open-source ecosystem. What you will do Release our 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 tooling and services: both internal facing (internal research) and external facing (open-source libraries) Implement and optimize open-source and internal libraries for performance and accuracy, ensuring production readiness and employing cutting-edge technology and innovative approaches Collaborate with the open-source community (PyTorch, vLLM, Hugging Face) About You Master’s degree in Computer Science, Machine Learning, Data Science, or a related field Experience contributing to popular open-source libraries such as PyTorch, Tensorflow, JAX, vLLM, Transformers, Llama.cpp, … Passion for contributing to the open-source software ecosystem Expert programming skills in Python, PyTorch, MLOps Adaptable, proactive, and autonomous Attention to detail and a drive to go the last mile to build almost perfect tools Deep understanding of machine learning approaches, especially LLMs and algorithms Low-ego, collaborative and have a real team player mindset Preferred Experience Experience with training and fine-tuning large language models (e.g., distillation, supervised fine-tuning, policy optimization) Experience working with Slurm Worked with research teams before Experience as a core-maintainer of a popular ML open-source library Location & Remote This role is primarily based at one of our European offices (Paris and London). Candidates should reside in or be willing to relocate to France or the UK. In certain specific situations, we will also consider remote candidates based in France or the UK. Onboarding: first month onsite with accommodation and travel covered; thereafter at least 3 days per month onsite. What we offer Competitive salary and equity Health insurance Transportation allowance Sport allowance Meal vouchers Private pension plan Generous parental leave policy Visa sponsorship Recruitment Note We may use artificial intelligence tools to support parts of the hiring process. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us. #J-18808-Ljbffr