JAX Jobs
67 open roles, updated daily
Filters2
Join Waymo as a Summer Intern in Software Engineering for Sys Intel & Machine Learning in Mountain View, CA. Help optimize autonomous driving models.
Join Waymo as a Summer Intern in Software Engineering for Sys Intel & Machine Learning in Mountain View, CA. Help optimize autonomous driving models.
Lead generative chemistry AI models at Flagship Pioneering. Drive multi-modal drug discovery, manage computational teams, and shape venture strategy.
Lead generative chemistry AI models at Flagship Pioneering. Drive multi-modal drug discovery, manage computational teams, and shape venture strategy.
Join Waymo as a PhD Software Engineering Intern in Simulation. Work on uncertainty-aware algorithms in a hybrid role based in Mountain View.
Join Waymo as a PhD Software Engineering Intern in Simulation. Work on uncertainty-aware algorithms in a hybrid role based in Mountain View.
Join Waymo as a Machine Learning Engineering Intern in San Francisco. Train multi-task transformers on TPUs for autonomous driving.
Join Waymo as a Machine Learning Engineering Intern in San Francisco. Train multi-task transformers on TPUs for autonomous driving.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
Join Writer as an AI Engineer in San Francisco, New York, or Seattle. Build scalable enterprise LLMs and AI agents with a $198K-$259K salary range.
Join Writer as an AI Engineer in San Francisco, New York, or Seattle. Build scalable enterprise LLMs and AI agents with a $198K-$259K salary range.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.
Join Waymo as a 2027 Summer Intern in Simulation Evaluation ML. Build semantic extraction pipelines and temporal graph models for autonomous driving.
Join Waymo as a 2027 Summer Intern in Simulation Evaluation ML. Build semantic extraction pipelines and temporal graph models for autonomous driving.
Join Thinking Machines in San Francisco as a Research Mid Training professional. Shape core AI capabilities with a salary up to $475,000.
Join Thinking Machines in San Francisco as a Research Mid Training professional. Shape core AI capabilities with a salary up to $475,000.
Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.
Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.
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 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.