BLACK FRIDAY - Get 50% off new job postings now through Dec 31 using code BLACK50

JAX jobs

Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.

405
open roles
380
posted this week
156
fully remote
144
pay $150k+
Dyna Robotics
Exceptional Software Engineer
Dyna Robotics - Redwood City, CA - On-site
$180k–400k4 days ago
Dyna Robotics
Research Scientist, Simulation
Dyna Robotics - Redwood City, CA - On-site
4 days ago
Dyna Robotics
Applied Researcher - Deployment Intelligence
Dyna Robotics - Redwood City, CA - On-site
4 days ago
Dyna Robotics
Research Internship
Dyna Robotics - Redwood City, CA - On-site
4 days ago
Dyna Robotics
Research Technical Program Manager, Robotics
Dyna Robotics - Redwood City, CA - On-site
4 days ago
Dyna Robotics
Research Scientist
Dyna Robotics - Redwood City, CA - On-site
$200k–400k4 days ago
Matter Intelligence
Data Infra - Signals and Evaluations (IC)
Matter Intelligence - San Francisco, CA - On-site
4 days ago
Mind Robotics
Research Engineer
Mind Robotics - Palo Alto, CA - On-site
4 days ago
Periodic Labs
Computational Scientist, Differentiable Physics
Periodic Labs - Menlo Park, CA - On-site
$250k–350k4 days ago
Inferact
Inference Engineering, Co-op
Inferact - San Francisco, CA - On-site
4 days ago
Inferact
Member of Technical Staff, TPU Kernel Engineering
Inferact - San Francisco, CA - On-site
$200k–400k4 days ago
Eight Sleep
Machine Learning Engineer (Foundation Models & Personalization)
Eight Sleep - San Francisco, CA - On-site
4 days ago
SandboxAQ
Machine Learning Scientist - Molecular Force Fields
SandboxAQ - United States - Hybrid
$134k–252k4 days ago
SandboxAQ
Computational Chemist - Machine Learned Force Fields, Material Discovery
SandboxAQ - United States - Hybrid
$134k–252k4 days ago
SandboxAQ
Machine Learning Scientist, Generative Molecular Design (Material Discovery)
SandboxAQ - United States - Hybrid
$134k–252k4 days ago
SandboxAQ
Machine Learning Scientist – Generative Molecular Design
SandboxAQ - United States - Hybrid
$134k–252k4 days ago
SandboxAQ
Senior Research Scientist, Magnets
SandboxAQ - United States - Hybrid
$134k–252k4 days ago
SandboxAQ
Senior Director of Product Management, Materials Science Software
SandboxAQ - United States - Remote
$189k–354k4 days ago
SandboxAQ
Senior Machine Learning Engineer, AI Generation Engine
SandboxAQ - United States - Remote
$131k–246k4 days ago
SandboxAQ
Director of Product Management, Financial Simulation and Modeling
SandboxAQ - United States - Remote
$163k–306k4 days ago
SandboxAQ
Staff Machine Learning Engineer, AI Generation Engine
SandboxAQ - United States - Remote
$157k–294k4 days ago
SandboxAQ
Senior Machine Learning Engineer, AI Generation Engine
SandboxAQ - Canada - Remote
4 days ago
SandboxAQ
Staff Machine Learning Engineer, AI Generation Engine
SandboxAQ - Canada - Remote
4 days ago
SandboxAQ
Director of Product Management, Financial Simulation and Modeling
SandboxAQ - Canada - Remote
4 days ago
Tessera Labs
Research Engineer
Tessera Labs - San Jose, CA +1 - Hybrid
$200k–300k4 days ago