Latest AI JobsBrowse Jobs

Jax Jobs

159 open roles, updated daily

Filters2
159 jobs foundClear all
W
Research Scientist, World Model Post-Training
Waymo
28d ago

Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.

Mountain View, CA, USA; San Francisco, CA, USA; New York, NY, USAAI Foundations (7SJ)Full-timeSenior · 3+ yrsOpen · checked 9d ago
PyTorchTensorFlowJAX
$213K — $263K USDView Job
MoneyLion logo
Senior Manager, Applied AI Research
MoneyLion
28d ago

Apply for the Senior Manager, Applied AI Research role at MoneyLion in Prague or remote. Lead AI initiatives protecting 500 million users worldwide.

CZE - PragueTechnologyFull-timeRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
Salary not listedView Job
Discord logo
Staff Software Engineer, Machine Learning (Consumer Revenue)
Discord
29d ago

Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.

San Francisco Bay Area or Remote (U.S.)Product EngineeringFull-timeSenior · 8+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$272K to $340KView Job
Thinking Machines logo
Research, Mid Training
Thinking Machines
29d ago

Join Thinking Machines in San Francisco as a Research Mid Training professional. Shape core AI capabilities with a salary up to $475,000.

San FranciscoResearchFull-timeOpen · checked 9d ago
PythonPyTorchTensorFlow
$350K- $475K USDView Job
sandboxaq logo
Senior Research Scientist, Magnets
sandboxaq
29d ago

Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.

United StatesAI Simulation Full-timeSenior · 3+ yrsRemoteOpen · checked 9d ago
PythonPyTorchJAX
$134.4K – $252KView Job
F
Machine Learning Engineer (m/f/d)
focused
29d ago

Join Focused Energy in Darmstadt, DE as a Machine Learning Engineer. Build ML models for laser-driven inertial fusion and digital twin environments.

Darmstadt, DESoftwareFull-timeSeniorRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
Salary not listedView Job
F
Machine Learning Engineer
focused
29d ago

Join Focused Energy in Austin, TX as a Machine Learning Engineer. Build models for laser-driven fusion technology. View benefits and apply.

Austin, TXSoftwareFull-timeSeniorRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
Salary not listedView Job
Mistral AI logo
Applied Scientist (Internship in Paris or London)
Mistral AI
32d ago

Mistral AI seeks Applied Scientist Interns based in Korea for Paris or London, leading to a full-time Seoul role. Apply now on their careers page.

SeoulSolutionsFull-timeOpen · checked 9d ago
PythonPyTorchJAX
Salary not listedView Job
W
Machine Learning Engineer, Driver Refinement & Validation, Perception
Waymo
32d ago

Join Waymo in Mountain View or San Francisco as a Machine Learning Engineer. Build autonomous driving tech with PyTorch or JAX. Salary: $175K-$215K.

Mountain View, CA, USA; San Francisco, CA, USAPerception (7LT)Full-timeSenior · 3+ yrsOpen · checked 9d ago
PythonPyTorchJAX
$175K — $215K USDView Job
P
Senior Simulation Engineer, Computational Physics
proxima-fusion
32d ago

Apply for the Senior Simulation Engineer role at Proxima Fusion in Munich. Build advanced simulation tools for commercial fusion energy.

MunichStellarator Architecture, Plasma Modelling & Research, Numerical OptimizationFull-timeSeniorOpen · checked 9d ago
PyTorchJAXMachine Learning
Salary not listedView Job
Niantic Spatial logo
Senior Computer Vision Engineer (Localization)
Niantic Spatial
33d ago

Apply for the Senior Computer Vision Engineer role at Niantic Spatial in San Francisco or remote. Build physical AI mapping and VPS tech.

San Francisco, CAEngineeringFull-timeRemoteOpen · checked 9d ago
PythonPyTorchJAX
$189K – $255KView Job
W
Staff Tech Lead Manager, Machine Learning, Simulator Evaluation
Waymo
33d ago

Join Waymo as a Staff Tech Lead Manager for Machine Learning, Simulator Evaluation. Lead an applied ML team building evaluation tools for autonomous driving.

Mountain View, California, United States; San Francisco, California, United States. Simulation (7XW)Full-timeLead · 7+ yrsOpen · checked 9d ago
PythonPyTorchTensorFlow
$251K — $310K USDView Job
Thinking Machines logo
Research, Finetuning Science
Thinking Machines
34d ago

Apply for the Research, Finetuning Science 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
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
34d 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
Staff Machine Learning Engineer
watney
34d ago

Apply for the Staff Machine Learning Engineer role at watney in San Francisco. Build models for autonomous robotic systems.

San FranciscoEngineeringFull-timeSeniorOpen · checked 9d ago
PythonPyTorchJAX
Salary not listedView Job
M
MACHINE LEARNING ENGINEER (GENERAL)
makermaker.ai
34d ago

Join makermaker.ai in San Francisco as a Senior Machine Learning Engineer to build production-grade infrastructure for autonomous research agents.

San FranciscoTechnical StaffFull-timeSenior · 6+ yrsOpen · checked 9d ago
PythonPyTorchJAX
Salary not listedView Job
Instacart logo
Senior Machine Learning Engineer, Ads Response Prediction
Instacart
35d 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
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
36d 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
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
39d 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