Applied Scientist: AI Optimisation & OR

Mistral; Lausanne; In loco; CHF 120.000 - 190.000; Tempo pieno

Descrizione del lavoro

Mistral AI seeks Applied Scientists and Research Engineers focused on optimisation and operational research. You will post-train SOTA large language models to formulate and solve optimisation problems, and apply them to diverse use cases.

Working cross-functionally with internal science, engineering and product teams, you will deliver high-impact AI solutions while tackling complex real-world problems in the industry.

Competenze

• Fluent English with excellent communication skills.
• Expert with PyTorch or JAX.
• Experience training or fine-tuning large language models for reasoning, code, or mathematics.
• Strong background in operational research or mathematical optimisation with hands-on experience using solvers (Gurobi, CPLEX, OR-Tools) and modelling languages (Pyomo, AMPL).
• Not afraid of contributing to a big codebase and working independently.
• Writes clean, readable, high-performance Python code.
• Moves independently and navigates uncertainty with confidence.
• Low-ego, collaborative and eager to learn.
• Track record of success through projects or academia.

Mansioni

• Run pre-training, post-training and deploy state of the art models on clusters with thousands of GPUs.
• Generate and curate datasets for optimisation modelling (LP, MILP, constraint programming, scheduling, routing).
• Use solvers as verifiers to build reward signals for reinforcement learning.
• Develop 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.

Conoscenze

English fluency

PyTorch

JAX

LLM fine-tuning

Operational research

Python

Solvers (Gurobi/CPLEX/OR-Tools)

Formazione

PhD / Master in Operational Research / Applied Mathematics / Machine Learning

Strumenti

Gurobi

CPLEX

OR-Tools

Pyomo

AMPL

Descrizione del lavoro

Mistral AI seeks Applied Scientists and Research Engineers focused on optimisation and operational research. You will post-train SOTA large language models to formulate and solve optimisation problems, and apply them to diverse use cases.

Working cross-functionally with internal science, engineering and product teams, you will deliver high-impact AI solutions while tackling complex real-world problems in the industry.