Applied Scientist

Job summary

Responsibilities

Mistral AI is seeking Applied Scientists and Research Engineers to drive innovative research and collaborate with clients on complex research projects.

• Run pre-training, post-training and deploy state of the art models on clusters with thousands of GPUs. You don’t panic when you see OOM errors or when NCCL feels like not wanting…
• Generate and curate data for pre-training and post-training, working on evaluations and making sure the model’s performance beats expectations.
• 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 using agents and RAG pipelines.
• Manage research projects and communications with client research teams.

What they're looking for

• You have excellent communication skills. You are at ease explaining complex technical concepts to both technical and non-technical audiences.
• You’re an expert with PyTorch or JAX.
• You’re not afraid of contributing to a big codebase and can find yourself around independently with little guidance.
• You write clean, readable, high-performance, fault-tolerant Python code.
• You don’t need roadmaps: you just do. You don’t need a manager: you just ship.

Tech stack

• RAG
• PyTorch
• JAX
• Python