JAX Jobs
68 open roles, updated daily
Filters2
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.
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 Focused Energy in Darmstadt, DE as a Machine Learning Engineer. Build ML models for laser-driven inertial fusion and digital twin environments.
Join Focused Energy in Darmstadt, DE as a Machine Learning Engineer. Build ML models for laser-driven inertial fusion and digital twin environments.
Join Focused Energy in Austin, TX as a Machine Learning Engineer. Build models for laser-driven fusion technology. View benefits and apply.
Join Focused Energy in Austin, TX as a Machine Learning Engineer. Build models for laser-driven fusion technology. View benefits and apply.
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 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.
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.
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.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
Apply for the Member of Technical Staff - Machine Learning role at inductive-bio in New York City, San Francisco, or Boston. Salary $180K–$250K + equity.
Apply for the Member of Technical Staff - Machine Learning role at inductive-bio in New York City, San Francisco, or Boston. Salary $180K–$250K + equity.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join Isomorphic Labs in London as a Research Engineer for LLM Performance. Scale foundational AI models for drug discovery using JAX, PyTorch, and distributed training.
Join Isomorphic Labs in London as a Research Engineer for LLM Performance. Scale foundational AI models for drug discovery using JAX, PyTorch, and distributed training.
Join Maincode in Melbourne as a Forward-Deployed AI Engineer. Deploy local AI stacks for justice and community safety customers. Apply today.
Join Maincode in Melbourne as a Forward-Deployed AI Engineer. Deploy local AI stacks for justice and community safety customers. Apply today.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.