Latest AI JobsBrowse Jobs

Jax Jobs

159 open roles, updated daily

Filters2
159 jobs foundClear all
inductive-bio logo
Member of Technical Staff - Machine Learning
inductive-bio
39d 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
deepjudge logo
Machine Learning Engineer
deepjudge
40d ago

Apply for the Machine Learning Engineer role at DeepJudge in Zurich HQ. Build and scale enterprise AI systems for legal search products.

Zurich HQEngineeringFull-timeMidOpen · checked 9d ago
PyTorchTensorFlowJAX
Salary not listedView Job
W
Senior Machine Learning Engineer, Predictive Planner, Systems Arch
Waymo
40d 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
Mistral AI logo
Compute Solution Architect
Mistral AI
41d ago

Apply for the Compute Solution Architect role at Mistral AI in New York. Support strategic AI infrastructure customers and scale GPU workloads.

New YorkSolutionsFull-timeRemoteOpen · checked 9d ago
PyTorchTensorFlowJAX
Salary not listedView Job
Anthropic logo
Staff+ Software Engineer, ML Inference Path
Anthropic
41d 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
42d 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
42d 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
Isomorphic Labs logo
Research Engineer (LLM Performance), London
Isomorphic Labs
43d ago

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.

LondonML ResearchFull-timeSeniorOpen · checked 9d ago
PyTorchJAXLLM
Salary not listedView Job
M
Forward-Deployed AI Engineer, Justice and Community Safety
maincode
44d ago

Join Maincode in Melbourne as a Forward-Deployed AI Engineer. Deploy local AI stacks for justice and community safety customers. Apply today.

MelbourneForward Deployed EngineeringFull-timeSeniorOpen · checked 9d ago
PythonPyTorchJAX
Salary not listedView Job
Cohere logo
Senior Software Engineer, GPU Infrastructure (HPC)
Cohere
47d 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
Mistral AI logo
Research Engineer, Machine Learning
Mistral AI
111d ago

Join Mistral AI in Paris as a Research Engineer, Machine Learning. Build and optimize large-scale learning systems for open-weight models.

ParisScienceFull-timeOpen · checked 9d ago
PythonPyTorchTensorFlow
Salary not listedView Job