Jax Jobs
41 open roles, updated daily
Filters3
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 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 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.
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.
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 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 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.
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 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 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.
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.
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.
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.