Jax jobs in AI
37 active AI roles hiring for Jax, 6 of them remote.
- Machine Learning Engineer, PerceptionWaymoMountain View, CA, USA; San Francisco, CA, USAFull-time
- 2027 Summer Intern, PhD, Learning-Based Behavior for Special VehiclesWaymoMountain View, CaliforniaFull-time
- 2027 Summer Intern, MS/PhD, Sim-Realism ML InfrastructureWaymoLondon, England, United KingdomFull-time
- 2027 Summer Intern, MS/PhD, Machine Learning, Simulation RealismWaymoLondon, England, United KingdomFull-time
- Staff Product Manager, Training Infrastructure - Weights & BiasesCoreWeaveLivingston, NJ / New York, NY / San Francisco, CA / Sunnyvale, CA / Bellevue, WA Full-time$188,000 - $275,000/yr
- Research Intern, Model Shaping (Winter 2027)Together AISan Francisco, AmsterdamFull-time$58,000 - $70,000/yr
- Research Intern, Model Shaping (Summer 2027)Together AISan Francisco, AmsterdamFull-time$58,000 - $70,000/yr
- Research Intern, Inference (Winter 2027)Together AISan FranciscoFull-time$58,000 - $70,000/yr
- Research Intern, Inference (Summer 2027)Together AISan FranciscoFull-time$58,000 - $70,000/yr
- Research Intern, Frontier Agents (Winter 2027)Together AISan FranciscoFull-time$58,000 - $70,000/yr
- Research Intern, Frontier Agents (Summer 2027)Together AISan FranciscoFull-time$58,000 - $70,000/yr
- Global Capacity Manager - TPU FocusBasetenRemoteSan FranciscoFull-time
- Senior Machine Learning Engineer, Predictive Planner, Systems ArchWaymoMountain View, CA, USA; San Francisco, CA, USAFull-time
- Research Scientist, Map ScalabilityWaymoMountain View, CA, U.S.A ; San Francisco, CA, U.S.A ; New York City, NY, U.S.AFull-time
- 2027 Summer Intern, PhD, Vehicle Intent and PredictionWaymoMountain View, CaliforniaFull-time
- 2027 Summer Intern, PhD, Machine Learning, Computer VisionWaymoMountain View, CaliforniaFull-time
- Member of Technical Staff (ML)RekaRemoteUK, USAFull-time
- Senior Machine Learning Engineer, Multimodal PerceptionWaymoMountain View, California, USAFull-time
- ML Infra Engineer, ModelingPhysical IntelligenceSan FranciscoFull-time
- Sr Staff Tech Lead, Machine Learning Engineer, PerceptionWaymoMountain View, California, United States, San Francisco, California, United StatesFull-time
- Senior Machine Learning Engineer, Multimodal PerceptionWaymoMountain View, California, USAFull-time
- Senior Research EngineerAssemblyAIRemoteRemote - New YorkFull-time$270,000 - $310,000/yr
- Forward Deployed Engineer (Training)BasetenRemoteSan FranciscoFull-time
- Staff Research Scientist, Foundation Models RecipesWaymoMountain View, CA, US; San Francisco, CA, US; Kirkland, WA, USFull-time
- Staff ML Engineer, Perception ResearchWaymoMountain View, CA, USA ; New York City, NY, USA ; Kirkland, WA, USA ; San Francisco, CA, USAFull-time
- Software Engineer, Compute SystemsWaymoTaipei, Taiwan; Hsinchu, TaiwanFull-time
- Principal AI Systems Performance EngineerSambaNovaSan Jose, California, United StatesFull-time
- Member of Technical Staff, Agentic EnvironmentsCohereRemoteEuropeFull-time
- Staff Machine Learning Engineer - Vision-Language Foundation ModelsWaymoMountain View, California, United StatesFull-time
- Perception Machine Learning Engineer - Continuous LearningWaymoMountain View, CA, USA; San Francisco, CA, USAFull-time
- ML Engineer, Foundation Model RecipesWaymoMountain View, CA, USA; San Francisco, CA, USA; Kirkland, WA, USA; New York, NY, USAFull-time
- Helix AI Engineer, Training PerformanceFigure AISan Jose, CAFull-time$200,000 - $400,000/yr
- Senior ML Engineer (AI Research, Physical AI)NebiusRemoteEurope, Netherlands, UKFull-time
- ML Engineer, Foundation Model InfrastructureWaymoMountain View, CA, USA; San Francisco, CA, USA; Kirkland, WA, USA; New York, NY, USAFull-time
- Senior Research Scientist, AI Foundations RecipesWaymoOxford, UK ; London, UKFull-time
- Engineering Manager, GPU InfrastructureCohereUnited StatesFull-time
Showing 36 of 37 open roles. Browse all AI jobs.