Latest AI JobsBrowse Jobs

Jax Jobs

191 open roles, updated daily

Filters1
191 jobs foundClear all
M
Data Infra - Signals and Evaluations (IC)
matter-intelligence
36d ago

Apply for the Data Infra - Signals and Evaluations role at Matter Intelligence in San Francisco. Build datasets, benchmarks, and evaluation workflows.

San FranciscoSoftware EngineeringFull-timeOpen · checked 9d ago
PythonPyTorchJAX
Salary not listedView Job
phantom logo
Staff Machine Learning Engineer
phantom
37d ago

Apply for the Staff Machine Learning Engineer role at Phantom. Remote position involving growth ML initiatives, system architecture, and team mentorship.

RemoteEngineeringFull-timeStaff · 8+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
Salary not listedView Job
P
Member of Technical Staff - ML Infrastructure Engineer, Post-training
Preference-Model
40d 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
40d 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
40d 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
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
B
2027 Internship Onboard Infrastructure Engineer, ML Inference
bedrock-robotics
41d ago

Apply for the 2027 Internship Onboard Infrastructure Engineer, ML Inference role at bedrock-robotics in San Francisco, CA.

San Francisco, CAEngineeringInternshipInternOpen · checked 9d ago
PyTorchJAXLLM
Salary not listedView 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
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
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
43d 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
43d 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
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
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
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