JAX Jobs
191 open roles, updated daily
Filters1
Mistral AI seeks Applied Scientist Interns based in Korea for Paris or London, leading to a full-time Seoul role. Apply now on their careers page.
Mistral AI seeks Applied Scientist Interns based in Korea for Paris or London, leading to a full-time Seoul role. Apply now on their careers page.
Apply for the 2027 Summer Intern, PhD, Learning-Based Behavior for Special Vehicles role at Waymo in Mountain View, California.
Apply for the 2027 Summer Intern, PhD, Learning-Based Behavior for Special Vehicles role at Waymo in Mountain View, California.
Join Together AI as a Research Intern for Frontier Agents in San Francisco. Work on agentic AI, alignment, and scaling with a Ph.D.
Join Together AI as a Research Intern for Frontier Agents in San Francisco. Work on agentic AI, alignment, and scaling with a Ph.D.
Join Waymo in Mountain View or San Francisco as a Machine Learning Engineer. Build autonomous driving tech with PyTorch or JAX. Salary: $175K-$215K.
Join Waymo in Mountain View or San Francisco as a Machine Learning Engineer. Build autonomous driving tech with PyTorch or JAX. Salary: $175K-$215K.
Join Waymo in London as a Sim-Realism ML Infrastructure Intern. Work on distributed systems, Python, C++, and autonomous driving technology.
Join Waymo in London as a Sim-Realism ML Infrastructure Intern. Work on distributed systems, Python, C++, and autonomous driving technology.
Join Together AI as a Research Intern for Winter 2027 in San Francisco. Work on frontier agentic systems, reinforcement learning, and LLM alignment.
Join Together AI as a Research Intern for Winter 2027 in San Francisco. Work on frontier agentic systems, reinforcement learning, and LLM alignment.
Join Together AI as a Research Intern for Winter 2027 in San Francisco or Amsterdam. Work on model shaping, post-training, and efficient training.
Join Together AI as a Research Intern for Winter 2027 in San Francisco or Amsterdam. Work on model shaping, post-training, and efficient training.
Join Together AI as a Research Intern for Model Shaping in San Francisco or Amsterdam. Work on post-training, efficient training, and foundation models.
Join Together AI as a Research Intern for Model Shaping in San Francisco or Amsterdam. Work on post-training, efficient training, and foundation models.
Join Together AI as a Winter 2027 Inference Research Intern in San Francisco. Build scalable serving systems for foundation models. Pay: $58-$70/hr.
Join Together AI as a Winter 2027 Inference Research Intern in San Francisco. Build scalable serving systems for foundation models. Pay: $58-$70/hr.
Join Together AI as a Summer 2027 Inference Research Intern in San Francisco. Build scalable serving systems for large foundation models.
Join Together AI as a Summer 2027 Inference Research Intern in San Francisco. Build scalable serving systems for large foundation models.
Apply for the Senior Simulation Engineer role at Proxima Fusion in Munich. Build advanced simulation tools for commercial fusion energy.
Apply for the Senior Simulation Engineer role at Proxima Fusion in Munich. Build advanced simulation tools for commercial fusion energy.
PhD research internship at XPENG focusing on predictive world models for autonomous driving and robotics using generative AI. Work on-site in Santa Clara.
PhD research internship at XPENG focusing on predictive world models for autonomous driving and robotics using generative AI. Work on-site in Santa Clara.
Apply for the Senior Computer Vision Engineer role at Niantic Spatial in San Francisco or remote. Build physical AI mapping and VPS tech.
Apply for the Senior Computer Vision Engineer role at Niantic Spatial in San Francisco or remote. Build physical AI mapping and VPS tech.
Join Waymo as a Staff Tech Lead Manager for Machine Learning, Simulator Evaluation. Lead an applied ML team building evaluation tools for autonomous driving.
Join Waymo as a Staff Tech Lead Manager for Machine Learning, Simulator Evaluation. Lead an applied ML team building evaluation tools for autonomous driving.
Join Waymo in London as a Machine Learning Simulation Realism Intern. Contribute to autonomous driving technology with Python and deep learning frameworks.
Join Waymo in London as a Machine Learning Simulation Realism Intern. Contribute to autonomous driving technology with Python and deep learning frameworks.
Apply for the Research, Finetuning Science role at Thinking Machines in San Francisco or Remote. Earn $350k-$475k with visa sponsorship.
Apply for the Research, Finetuning Science role at Thinking Machines in San Francisco or Remote. Earn $350k-$475k with visa sponsorship.
Lead the Fine-tuning Science team at Thinking Machines in San Francisco. Shape frontier customization techniques. Apply on their careers page.
Lead the Fine-tuning Science team at Thinking Machines in San Francisco. Shape frontier customization techniques. Apply on their careers page.
Apply for the Research, Tinker, RL Systems role at Thinking Machines in San Francisco or remote. Earn $350k-$475k with visa sponsorship.
Apply for the Research, Tinker, RL Systems role at Thinking Machines in San Francisco or remote. Earn $350k-$475k with visa sponsorship.
Apply for the Staff Machine Learning Engineer role at watney in San Francisco. Build models for autonomous robotic systems.
Apply for the Staff Machine Learning Engineer role at watney in San Francisco. Build models for autonomous robotic systems.
Join makermaker.ai in San Francisco as a Senior Machine Learning Engineer to build production-grade infrastructure for autonomous research agents.
Join makermaker.ai in San Francisco as a Senior Machine Learning Engineer to build production-grade infrastructure for autonomous research agents.
Apply for the 2027 Summer Intern, PhD, Machine Learning, Computer Vision role at Waymo in Mountain View, California.
Apply for the 2027 Summer Intern, PhD, Machine Learning, Computer Vision role at Waymo in Mountain View, California.
Join Relay in Raleigh, NC for a paid Summer 2027 AI/ML Software Engineering Internship. Build voice and AI technology for frontline workers.
Join Relay in Raleigh, NC for a paid Summer 2027 AI/ML Software Engineering Internship. Build voice and AI technology for frontline workers.
Join Inferact in San Francisco as an Inference Engineering Co-op. Work with the creators of vLLM on AI inference infrastructure. Apply via official site.
Join Inferact in San Francisco as an Inference Engineering Co-op. Work with the creators of vLLM on AI inference infrastructure. Apply via official site.
Join Instacart as a Senior ML Engineer on the Ads Response Prediction team. Remote in Canada (ON, AB, BC, NS). Base pay $180k-$190k CAD.
Join Instacart as a Senior ML Engineer on the Ads Response Prediction team. Remote in Canada (ON, AB, BC, NS). Base pay $180k-$190k CAD.