Applied Scientist/research Engineer - Palo Alto/nyc

Mistral AI Mistral AI · AI Frontier · Palo Alto, CA · Solutions

Mistral AI is seeking Applied Scientists and Research Engineers to drive innovative research and collaborate with clients on complex research projects. The role involves developing SOTA models across modalities like text, image, and speech, applying them to diverse use cases, and delivering high-impact AI solutions. Responsibilities include running pre-training and post-training on large GPU clusters, generating and curating data, developing tools for data generation, training, evaluation, and deployment, collaborating on agent and RAG pipelines, and managing research projects with clients.

What you'd actually do

  1. 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 to talk.
  2. Generate and curate data for pre-training and post-training, working on evaluations and making sure the model’s performance beats expectations.
  3. Develop the necessary tools and frameworks to facilitate data generation, model training, evaluation and deployment.
  4. Collaborate with cross-functional teams to tackle complex use cases using agents and RAG pipelines.
  5. Manage research projects and communications with client research teams.

Skills

Required

  • PyTorch or JAX
  • Python
  • English fluency
  • excellent communication skills

Nice to have

  • PhD / master in a relevant field
  • agents
  • multi-modality
  • robotics
  • diffusion
  • time-series
  • contributed to a large codebase
  • publications in top academic journals or conferences
  • improving existing code by fixing typing issues, adding tests and improving CI pipelines

What the JD emphasized

  • track record of success
  • track record of publications

Other signals

  • develop SOTA models
  • apply these models across a diverse set of use cases
  • deliver high-impact AI solutions