JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
406
open roles
383
posted this week
156
fully remote
145
pay $150k+
Latest openings
Filter in search →
Research Scientist, Discovery (Engineering & Applied Sciences)
Anthropic - San Francisco, CA +1 - Hybrid
$350k–500k5h ago
Member of Technical Staff, Forward Deployed Engineer
Odyssey - Palo Alto, CA +1 - Hybrid
$165k–508k16h ago
Sr. Model Engineer, Voice
SpaceXAI - Palo Alto, CA - On-site
$220k–357k2 days ago
Sr. Model Engineer, Imagine
SpaceXAI - Palo Alto, CA - On-site
$220k–357k2 days ago
2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable Simulation
Waymo - Mountain View, CA - Hybrid
$177k–177k2 days ago
2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)
Waymo - Mountain View, CA - Hybrid
$146k–177k2 days ago
Research Engineer, Robotics: Physics Simulation and Sim-to-Real
The Allen Institute for Artificial Intelligence - Seattle, WA - On-site
$129k–193k2 days ago
Co-op, AI for Protein Design
Lila Sciences - San Francisco, CA - On-site
2 days ago
Senior Security Engineer (AI Safety), London, Lausanne
Isomorphic Labs - Lausanne; London - On-site
3 days ago
Model Engineer, Multimodal Understanding
SpaceXAI - Palo Alto, CA - On-site
$160k–250k3 days ago
Machine Learning Researcher - Agentic Science
PostEra - Cambridge, MA - Remote
$200k–300k3 days ago
Senior Machine Learning Scientist
Tahoe Therapeutics - South San Francisco, CA - Hybrid
$200k–275k3 days ago
Senior Machine Learning Engineer
Tahoe Therapeutics - South San Francisco, CA - Hybrid
3 days ago
Staff / Principal Software Engineer
GenBio AI - Palo Alto, CA - On-site
$250k–350k3 days ago
Research Scientist Intern, AI Molecular Design (Winter/Spring)
GenBio AI - Palo Alto, CA - On-site
$96k–120k3 days ago
Director, Engineering
GenBio AI - Palo Alto, CA - On-site
$300k–400k3 days ago
Machine Learning Engineer - Enterprise
Boson AI - Toronto, ON - On-site
3 days ago
Internship - MScAC 2027
Boson AI - Toronto, ON - Hybrid
3 days ago
AI Scientist - Machine Learning (KR/US)
Gauss Labs - Palo Alto, CA +1 - On-site
3 days ago
Quantitative Researcher
QuantCo - Boston, MA +1 - Hybrid
3 days ago
Applied Scientist, AI/ML
QuantCo - Boston, MA +1 - Hybrid
3 days ago
Senior Program Scientist, AI and Advanced Computing Institute
Schmidt Entities - New York, NY - On-site
$300k–325k3 days ago
Fellow in Residence
Schmidt Entities - New York, NY - On-site
$160k–160k3 days ago
AI Research Scientist
Collate - San Francisco, CA - On-site
$188k–215k3 days ago
Principal Architect, AI/ML
Zencore - EMEA (Remote) - Remote
3 days ago