Skills · Make it fast and big
JAX
Course coming soon
Functional transforms, jit and sharding on TPUs and GPUs.
Mentioned in 92 of 338 recorded research roles (27%) and 167 of 1,837 technical roles (9%), as of October 2, 2026.
Learn it here
The course page has its planned syllabus and a sign-up, which tells us how many people want it.
These counts describe the October 2, 2026 snapshot; postings need a fresh check. Explore opportunities.
Everything here that teaches it
Nothing on this site teaches it yet. It is listed so you can see how often labs ask for it, and the course in preparation above has a sign-up.
Also in Make it fast and big: Inference and serving, PyTorch, Distributed training, GPU kernels.
Who asks for it
- Research scientists and engineers 92 of 338
- Fellowships, residencies and internships 11 of 44
- Applied AI and forward-deployed 1 of 323
- ML systems and infrastructure 32 of 109
- Machine learning engineers 9 of 146
- Software engineering at labs 22 of 877
Recorded roles that mention it
167 recorded postings mention it. Here are the newest 60 research, fellowship, applied and ML-systems roles; the rest are on the role pages linked above. Each title opens the lab's own posting.
- AI Research Engineer, Post-Training ↗
Lovable · Research scientists and engineers · No degree stated
- Machine Learning Scientist ↗
Relace · Research scientists and engineers · Master's, PhD or equivalent
- Software Engineer, Distributed Systems ↗
Thinking Machines Lab · ML systems and infrastructure · No degree stated · $300k to $350k base
- Software Engineer, Inference ↗
Thinking Machines Lab · ML systems and infrastructure · No degree stated · $300k to $350k base
- Research Engineer, Asta ↗
Allen Institute for AI (Ai2) · Research scientists and engineers · PhD preferred · $129k to $193k base
- Research Engineer, Climate Modeling ↗
Allen Institute for AI (Ai2) · Research scientists and engineers · PhD preferred · $147k to $220k base
- Research Scientist, Climate Modeling ↗
Allen Institute for AI (Ai2) · Research scientists and engineers · PhD required · $167k to $261k base
- Senior Research Engineer, Olmo + Molmo ↗
Allen Institute for AI (Ai2) · Research scientists and engineers · PhD or equivalent experience · $174k to $261k base
- Performance Engineer, GPU ↗
Anthropic · ML systems and infrastructure · No degree stated · $280k to $850k base
- Performance Engineer, Inference Systems ↗
Anthropic · ML systems and infrastructure · No degree stated · $350k to $850k base
- Research Engineer / Research Scientist, RL Frontiers ↗
Anthropic · Research scientists and engineers · No degree stated · $500k to $850k base
- Research Engineer, Code RL (Reinforcement Learning) ↗
Anthropic · Research scientists and engineers · No degree stated · $500k to $850k base
- Research Engineer, Discovery ↗
Anthropic · Research scientists and engineers · No degree stated · $350k to $850k base
- Research Engineer, Interpretability ↗
Anthropic · Research scientists and engineers · No degree stated · $315k to $560k base
- Research Engineer, Machine Learning (Reinforcement Learning) ↗
Anthropic · Research scientists and engineers · No degree stated · $500k to $850k base
- Research Engineer, Machine Learning (Reinforcement Learning) ↗
Anthropic · Research scientists and engineers · No degree stated
- Research Engineer, Machine Learning (RL Velocity) ↗
Anthropic · Research scientists and engineers · No degree stated
- Research Engineer, Machine Learning (RL Velocity) ↗
Anthropic · Research scientists and engineers · No degree stated · $500k to $850k base
- Research Engineer, Pretraining Scaling ↗
Anthropic · Research scientists and engineers · No degree stated · $350k to $850k base
- Research Engineer, Pretraining Scaling - London ↗
Anthropic · Research scientists and engineers · No degree stated
- Research Engineer, RL Engineering ↗
Anthropic · Research scientists and engineers · No degree stated · $500k to $850k base
- Staff Software Engineer, Inference ↗
Anthropic · ML systems and infrastructure · No degree stated
- Staff Software Engineer, Infrastructure (Distributed Systems) ↗
Anthropic · ML systems and infrastructure · No degree stated
- Staff+ Software Engineer, Inference Velocity ↗
Anthropic · ML systems and infrastructure · No degree stated · $405k to $485k base
- Staff+ Software Engineer, Infrastructure (Distributed Systems) ↗
Anthropic · ML systems and infrastructure · No degree stated · $320k to $485k base
- Staff+ Software Engineer, ML Inference Path ↗
Anthropic · ML systems and infrastructure · No degree stated · $320k to $485k base
- TPU Kernel Engineer ↗
Anthropic · ML systems and infrastructure · No degree stated · $280k to $850k base
- Applied Researcher, Audio ↗
Cartesia · Research scientists and engineers · No degree stated · $200k to $350k base
- Machine Learning Infrastructure Engineer ↗
Character.AI · ML systems and infrastructure · No degree stated · $150k to $350k base
- Lead Member of Technical Staff, Inference Infrastructure ↗
Cohere · ML systems and infrastructure · No degree stated · $235k to $325k base
- Machine Learning Intern/Co-op (Winter 2027) ↗
Cohere · Fellowships, residencies and internships · No degree stated
- Member of Technical Staff, Data Analysis and Evaluation ↗
Cohere · Research scientists and engineers · No degree stated
- Member of Technical Staff, Integration/RL Team (Research Engineer) ↗
Cohere · Research scientists and engineers · No degree stated
- Member of Technical Staff, Modeling ↗
Cohere · Research scientists and engineers · No degree stated
- Member of Technical Staff, Post-Training ↗
Cohere · Research scientists and engineers · No degree stated
- Member of Technical Staff, Training Infra Engineer ↗
Cohere · ML systems and infrastructure · No degree stated
- Member of Technical Staff, Training Performance Engineer ↗
Cohere · ML systems and infrastructure · No degree stated
- Research Internship (Winter 2027) ↗
Cohere · Fellowships, residencies and internships · PhD required
- Senior Member of Technical Staff, Multimodal AI ↗
Cohere · Research scientists and engineers · No degree stated
- Senior Member of Technical Staff, Safety and Security for Agents ↗
Cohere · Research scientists and engineers · No degree stated
- Senior ML Systems Engineer, Frameworks & Tooling ↗
Cohere · ML systems and infrastructure · No degree stated
- Senior Software Engineer, GPU Infrastructure (HPC) ↗
Cohere · ML systems and infrastructure · No degree stated
- Site Reliability Engineer, Inference Infrastructure ↗
Cohere · ML systems and infrastructure · No degree stated · $160k to $260k base
- Staff Software Engineer, Inference Infrastructure ↗
Cohere · ML systems and infrastructure · No degree stated · $190k to $325k base
- Post-training Agentic Research Scientist, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD or equivalent experience
- Research Engineer, Gemini Omni, DeepMind ↗
Google DeepMind · Research scientists and engineers · No degree stated
- Research Engineer, GenAI, Information Tasks, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Research Engineer, Generative Media, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Research Engineer, Responsible Frontier AI Research, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Research Engineer, Responsible Frontier AI Research, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Research Engineer, World Models, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Research Scientist, Agents, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD or equivalent experience
- Research Scientist, Multilingual, Multicultural and Multimodal LLM, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD or equivalent experience
- Research Scientist/Engineer, Autonomous Security, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD or equivalent experience
- Research Scientist/Engineer, Frontier Reasoning, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Senior Research Engineer, Embodied Generalist Agent, Gaming, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD preferred
- Senior Staff Research Scientist, Recommendation Systems, DeepMind ↗
Google DeepMind · Research scientists and engineers · PhD or equivalent experience
- Software Engineer, Model Inference, DeepMind ↗
Google DeepMind · ML systems and infrastructure · Bachelor's or equivalent
- Staff Software Engineer, Inference Performance Optimization, GenAI, DeepMind ↗
Google DeepMind · ML systems and infrastructure · Bachelor's or equivalent
- Research Engineer / Scientist, Post-training & Reinforcement Learning - London ↗
H Company · Research scientists and engineers · PhD preferred
