The Role
Mistral AI is seeking Applied Scientists and Research Engineers focused on formal methods. You will design, train, and deploy SOTA large language models and apply them to diverse use cases.
You'll work cross-functionally with internal and external science, engineering, and product teams to deliver high-impact AI solutions.
What You Will Do
Run pre-training, post-training and deploy state of the art models on large clusters.
Generate and curate datasets for theorem proving and software verification, and build evaluators and benchmarks for formal reasoning.
Develop the necessary tools and frameworks to facilitate data generation, model training, evaluation and deployment.
Collaborate with cross-functional teams to tackle complex use cases.
Manage research projects and communications with client research teams.
What We're Looking For
Fluent in English with excellent communication skills — at ease explaining complex technical concepts to both technical and non-technical audiences.
Expert with PyTorch or JAX.
Not afraid of contributing to a big codebase; finds their way around independently with little guidance.
Writes clean, readable, high-performance, fault-tolerant Python code.
Doesn’t need roadmaps: moves independently and navigates uncertainty with confidence.
Low-ego, collaborative and eager to learn.
Track record of success through personal projects, professional projects or academia.
Nice to have
PhD / Master in a relevant field (Computer Science, Mathematics, Machine Learning).
Experience with autoformalization, proof automation, or verified code generation.
Experience with reinforcement learning using verifiable rewards, ideally from Lean or other proof assistants.
Contributions to a large open-source or industry codebase.
Publications in top academic journals or conferences.
Enjoys improving existing code: typing, tests, CI pipelines.