Jobs using JAX
157 open · Open roles whose own postings name JAX
The newest
- GPU Kernel Engineer · Sciforium · San Francisco, CA
- Sr. Machine Learning Engineer, Autonomy · 42dot · Sunnyvale, United States
- 2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable Simulation · Waymo · Mountain View, California, United States
- 2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model) · Waymo · Mountain View, California, United States
- Applied AI Engineer · Bjak · United States
- Senior ML Engineer, Agentic AI · Ellipsis Health · San Francisco - Hybrid
- Member of Technical Staff - Research Intern · Trajectory · San Francisco
- Applied Researcher · River AI Inc. · Palo Alto, CA
- Staff Machine Learning Engineer, AI Infrastructure · Waymo · Mountain View, CA, USA; San Francisco, CA, USA; New York, NY, USA
- Senior Software Engineer · LanceDB · Americas or APAC
- Machine Learning Scientist, Biomolecular Structure Prediction · Profluent · Emeryville, California, United States
- Senior Machine Learning Engineer (Growth) · Discord · San Francisco Bay Area
- Senior Staff Machine Learning Engineer, Perception, Simulation Realism · Waymo · Mountain View, California, United States; San Francisco, California, United States
- AI engineer · Writer · San Francisco, CA
- Machine Learning Engineer Intern, AI & Robotics · RoboForce · Milpitas, CA
- Research Scientist – Multimodal Foundation Models & Visual Reasoning · Elorian · Palo Alto
- Research Engineer, Infrastructure · Harvey · San Francisco
- Research Engineer, Infrastructure · Harvey · New York
- Sr. Model Engineer, Post-Training & RL · SpaceXSI · Palo Alto, CA
- Senior Inference Optimization ML Engineer · Rhoda AI · Mountain View
- Staff Product Data Scientist, Marketplace ML · Waymo · Mountain View, CA, USA; San Francisco, CA, USA
- Senior Software Engineer, Machine Learning Platform · Airwallex · US - Seattle
- Machine Learning Engineer, Driving Foundation Models · 42dot · Sunnyvale, United States
- Software Engineer - Manipulation · Apptronik · Sunnyvale, CA
- Member of Technical Staff, AI Research · Physical Superintelligence · Boston
- AI Research Intern (BS/MS) · ATOMS · Mountain View, CA
- AI Research Intern (BS/MS) · Atoms · Mountain View, CA
- Research Scientist, AI · Pramaana Labs · Palo Alto
- Research Engineer · Pramaana Labs · Palo Alto
- GPU Kernel Engineer · Sciforium · San Francisco, CA
- Member of Technical Staff, Machine Learning · Bjak · United States
- Applied Machine Learning Engineer · Bjak · United States
- Robotics Engineering Intern — Hardware & Software · Trace · NYC
- 2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and Efficiency · Waymo · Mountain View, CA, USA
- Member of Technical Staff, Robotics Research · Sieve · San Francisco
- Senior Machine Learning Engineer, Scenes, Traffic Controls · Waymo · Mountain View, California, United States, San Francisco, California, United States
- Member of Technical Staff - Pre-Training Data · Reflection · San Francisco, CA
- 2027 Summer Intern, PhD, Research, Multimodal Foundation Models · Waymo · Mountain View, CA, USA
- Compiler Frontend Engineer · Normal Computing Corporation · Mountain View
- Software Engineer · Waymo · PERM - N/A