JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
404
open roles
379
posted this week
156
fully remote
143
pay $150k+
Page 3
Filter in search →
Software Engineer, ML Training Optimization
Zoox - Foster City, CA - On-site
$177k–270k3 days ago
Software Engineer Intern, Autonomy Software [Summer 2027]
Zoox - Foster City, CA - On-site
$78k–126k3 days ago
Senior/Staff Software Engineer, ML Training Optimization
Zoox - Foster City, CA +1 - On-site
$254k–350k3 days ago
Senior / Staff Machine Learning Engineer - Behavior Models for Road Users
Zoox - Foster City, CA - On-site
$254k–366k3 days ago
Engineering Manager, ML Performance Optimization
Zoox - Foster City, CA - On-site
$276k–343k3 days ago
Staff Machine Learning Engineer, Sensors & Intelligence
Jobgether - Canada - Remote
3 days ago
Staff Engineer, Hivemind Reasoning (R6292)
Shield AI - San Mateo, CA - On-site
$240k–360k3 days ago
Senior Manager, Engineering - Hivemind Reasoning (R6291)
Shield AI - San Mateo, CA - On-site
$290k–430k3 days ago
Founding Machine Learning Engineer
Orbit Neuro Co. - San Francisco, CA - On-site
$225k–275k3 days ago
Research Engineer
Lotus AI - San Francisco, CA - On-site
3 days ago
Machine Learning Engineer, Applied
Trace - New York, NY - Hybrid
$185k–245k3 days ago
Machine Learning Engineer (Physics-based)
Root Access - New York, NY - On-site
3 days ago
Applied Machine Learning
Ideogram - Toronto, ON +1 - On-site
3 days ago
Founding Engineer
Doji - New York, NY - On-site
3 days ago
Research Engineer
Doji - New York, NY - On-site
3 days ago
AI Engineer
Joyful Health - New York, NY - Hybrid
$150k–275k3 days ago
Member of Technical Staff - Machine Learning
Inductive Bio - New York, NY +2 - Hybrid
$180k–250k3 days ago
Member of Technical Staff - Software Engineering
Inductive Bio - New York, NY +2 - Hybrid
$180k–240k3 days ago
Member of Technical Staff, AI-Driven Compilation
San Francisco Tensor Company - San Francisco, CA - On-site
$275k–315k3 days ago
Member of Technical Staff, Post-Training & Applied Research
San Francisco Tensor Company - San Francisco, CA - On-site
$275k–315k3 days ago
Member of Technical Staff (Research Engineering)
ATG - New York, NY +1 - On-site
$180k–260k3 days ago
Research Scientist
ATG - New York, NY +1 - On-site
$200k–300k3 days ago
ML Research Scientist (MLRS) - Representation Learning for Molecular AI
Achira - San Francisco, CA +1 - Hybrid
$165k–259k3 days ago
ML Research Scientist (MLRS) - Generative AI
Achira - San Francisco, CA +1 - Hybrid
$165k–259k3 days ago
Machine Learning Research Engineer (MLRE) - Research
Achira - San Francisco, CA +1 - Hybrid
$165k–259k3 days ago