JAX Jobs
67 open roles, updated daily
Filters2
Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.
Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.
Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.
Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.
Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.
Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.
Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.
Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.
Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.
Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.
Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.
Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.
Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Research and implement world action models for autonomous driving at Waymo. Ideal for a Ph.D. candidate with a strong ML background.
Research and implement world action models for autonomous driving at Waymo. Ideal for a Ph.D. candidate with a strong ML background.
Develop ML solutions for lunar and Mars missions. 5-8 years experience required in autonomy, navigation, and mission planning.
Develop ML solutions for lunar and Mars missions. 5-8 years experience required in autonomy, navigation, and mission planning.
Join Waymo as a 2027 Summer Intern in Machine Learning for Driver Refinement Foundations. Work on LLMs, generative models, and autonomous driving tech.
Join Waymo as a 2027 Summer Intern in Machine Learning for Driver Refinement Foundations. Work on LLMs, generative models, and autonomous driving tech.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Join Anthropic's RL Scaling team to build next-gen algorithms and scale reinforcement learning for Claude. Annual salary $500,000 — $850,000 USD.
Join Anthropic's RL Scaling team to build next-gen algorithms and scale reinforcement learning for Claude. Annual salary $500,000 — $850,000 USD.
Join Waymo as a PhD Planner Machine Learning Intern in San Francisco. Research deep learning architectures for autonomous driving. Apply now.
Join Waymo as a PhD Planner Machine Learning Intern in San Francisco. Research deep learning architectures for autonomous driving. Apply now.
Join Waymo as a PhD Machine Learning Research Intern in Mountain View. Research generative models for autonomous driving planning and prediction.
Join Waymo as a PhD Machine Learning Research Intern in Mountain View. Research generative models for autonomous driving planning and prediction.
Join Ai2 in Seattle as a Research Engineer for a 3-year climate modeling project. Work with JAX, Google Cloud, and ML models. $146k-$220k.
Join Ai2 in Seattle as a Research Engineer for a 3-year climate modeling project. Work with JAX, Google Cloud, and ML models. $146k-$220k.
Join Waymo in Mountain View as a Perception Machine Learning Intern. Design multi-modality models using large-scale driving data.
Join Waymo in Mountain View as a Perception Machine Learning Intern. Design multi-modality models using large-scale driving data.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.