Latest AI JobsBrowse Jobs

Jax Jobs

67 open roles, updated daily

Filters2
67 jobs foundClear all
Thinking Machines logo
Research Lead, Tinker, Fine-tuning Science
Thinking Machines
34d ago

Lead the Fine-tuning Science team at Thinking Machines in San Francisco. Shape frontier customization techniques. Apply on their careers page.

San FranciscoResearchFull-timeOpen · checked 9d ago
PythonPyTorchTensorFlow
$350K - $475K USDView Job
Thinking Machines logo
Research, Tinker, RL Systems
Thinking Machines
35d ago

Apply for the Research, Tinker, RL Systems role at Thinking Machines in San Francisco or remote. Earn $350k-$475k with visa sponsorship.

San FranciscoResearchFull-timeOpen · checked 9d ago
PythonPyTorchTensorFlow
$350K - $475K USDView Job
W
2027 Summer Intern, PhD, Machine Learning, Computer Vision
Waymo
35d ago

Apply for the 2027 Summer Intern, PhD, Machine Learning, Computer Vision role at Waymo in Mountain View, California.

Mountain View, California, United StatesPipeline (N/A)InternshipInternshipOpen · checked 9d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
Instacart logo
Senior Machine Learning Engineer, Ads Response Prediction
Instacart
36d ago

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.

Canada - Remote (ON, AB, BC, or NS Only)Machine LearningFull-timeSenior · 3+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$180K — $190K CADView Job
P
Member of Technical Staff - ML Infrastructure Engineer, Post-training
Preference-Model
39d ago

Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.

San FranciscoEngineeringFull-timeSeniorOpen · checked 9d ago
PyTorchJAXLLM
$200K – $350KView Job
P
Member of Technical Staff - Research Engineer, Post-training
Preference-Model
39d ago

Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.

San FranciscoEngineeringFull-timeSeniorOpen · checked 9d ago
PythonPyTorchJAX
$200K – $350KView Job
P
Member of Technical Staff - Machine Learning Capabilities
Preference-Model
39d ago

Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.

San FranciscoEngineeringFull-timeSenior · 5+ yrsOpen · checked 9d ago
PythonPyTorchJAX
$200K – $350KView Job
Discord logo
Senior Software Engineer, Machine Learning (Safety)
Discord
40d ago

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.

San Francisco Bay AreaDevelopersFull-timeSenior · 4+ yrsOpen · checked 9d ago
PythonPyTorchTensorFlow
$220K to $275KView Job
inductive-bio logo
Member of Technical Staff - Machine Learning
inductive-bio
40d ago

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.

New York City, San Francisco, or BostonMachine LearningFull-timeSenior · 4+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$180K – $250KView Job
W
Senior Machine Learning Engineer, Predictive Planner, Systems Arch
Waymo
41d ago

Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.

Mountain View, CA, USA; San Francisco, CA, USAPlanner (7LU)Full-timeSenior · 4+ yrsOpen · checked 9d ago
PythonPyTorchTensorFlow
$213K — $263K USDView Job
W
2027 Summer Intern, PhD, Vehicle Intent and Prediction
Waymo
41d ago

Apply for the 2027 Summer Intern PhD Vehicle Intent and Prediction role at Waymo in Mountain View, CA. Hourly pay of $85 USD.

Mountain View, CaliforniaPipeline (N/A)InternshipInternshipOpen · checked 9d ago
PythonPyTorchJAX
$85 — $85 USDView Job
Anthropic logo
Staff+ Software Engineer, ML Inference Path
Anthropic
42d ago

Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.

San Francisco, CASafeguards (Trust & Safety)Full-timeSenior · 5+ yrsOpen · checked 9d ago
PythonPyTorchTensorFlow
$320K — $485K USDView Job
True Anomaly logo
Machine Learning Engineer (II-III), Space Edge Deployment
True Anomaly
42d ago

Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.

Denver, Colorado or Long Beach, CaliforniaAutonomy EngineeringFull-timeMidOpen · checked 9d ago
PythonPyTorchTensorFlow
$125K - $220KView Job
Thinking Machines logo
Research Software Engineer, Post Training
Thinking Machines
42d ago

Apply for the Research Software Engineer, Post Training role at Thinking Machines in San Francisco. Salary: $350k-$475k. Remote options available.

San FranciscoResearchFull-timeOpen · checked 9d ago
PythonPyTorchTensorFlow
$350K - $475K USDView Job
E
Thermo ML Resident
extropic
42d ago

Apply for the Thermo ML Resident role at Extropic in Boston. Flexible part-time or full-time machine learning residency paying $75K–$200K.

BostonApplications & AlgorithmsFull-timeEntryOpen · checked 9d ago
PythonPyTorchTensorFlow
$75K – $200KView Job
E
Thermo ML Resident
extropic
42d ago

Join Extropic as a Thermo ML Resident in San Francisco. Flexible part-time or full-time role with $75K-$200K salary.

San FranciscoApplications & AlgorithmsFull-timeEntryOpen · checked 9d ago
PythonPyTorchTensorFlow
$75K – $200KView Job
W
Research Scientist, Map Scalability
Waymo
43d ago

Apply for the Research Scientist, Map Scalability role at Waymo in Mountain View, San Francisco, or New York City. Salary $213,000 - $263,000 USD.

Mountain View, CA, U.S.A ; San Francisco, CA, U.S.A ; New York City, NY, U.S.AAI Foundations (7SJ)Full-timeMid · 1+ yrsOpen · checked 9d ago
PythonPyTorchTensorFlow
$213K — $263K USDView Job
Fireworks AI logo
Member of Technical Staff, Research (2026 PhD New Grad)
Fireworks AI
43d ago

Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.

San MateoEngineeringFull-timeRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$200K – $300KView Job
Cohere logo
Senior Software Engineer, GPU Infrastructure (HPC)
Cohere
48d ago

Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.

CanadaModelingFull-timeSeniorRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
CA$285K – CA$340KView Job