Latest AI JobsBrowse Jobs

Jax Jobs

191 open roles, updated daily

Filters1
191 jobs foundClear all
W
2027 Summer Intern, PhD, Research, AV Planning
Waymo
9d ago

Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Research, Post Training
Waymo
9d ago

Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.

Mountain View, CA, USA: San Francisco, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PyTorchJAXReinforcement Learning
$85 — $85 USDView Job
A
Machine Learning Engineer (2-5 yrs)
Advanced Space
9d ago

Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.

Westminster, COMachine LearningFull-timeMid · 2+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$100K -$134KView Job
G
Senior Machine Learning Engineer (Large Systems)
Graphcore
9d ago

Apply for the Senior Machine Learning Engineer role at Graphcore in Cambridge, Bristol, or London. Scale AI models on specialised hardware.

Cambridge, UKApplied AIFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
G
Senior Machine Learning Engineer (Large Systems)
Graphcore
9d ago

Apply for the Senior Machine Learning Engineer role at Graphcore in London, Bristol, or Cambridge. Scale AI models on specialized hardware.

London, UKApplied AIFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
G
Senior Machine Learning Engineer (Large Systems)
Graphcore
9d ago

Apply for the Senior Machine Learning Engineer role at Graphcore in Bristol, London or Cambridge. Scale AI models on specialized hardware.

Bristol, UKApplied AIFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
Cognition logo
Research Engineer, Safety & Alignment
Cognition
9d ago

Apply for the Research Engineer, Safety & Alignment role at Cognition in San Francisco. Build safety evals and alignment methods for AI agents.

San FranciscoResearch & DevelopmentFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
mach9 logo
Perception Intern
mach9
10d ago

Apply for the Perception Intern role at Mach9 in San Francisco. Work on 2D image and 3D point cloud models for geospatial data analysis.

San FranciscoEngineeringInternshipInternOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
W
2027 Summer Intern, PhD, Research, World Modeling Evaluation
Waymo
10d ago

Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Machine Learning, Planning/Prediction
Waymo
10d ago

Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
airwallex logo
Senior Software Engineer, Machine Learning Platform
airwallex
11d ago

Join Airwallex in Singapore as a Senior Software Engineer for ML Platform. Build risk decisioning infrastructure. On-site role (5 days/week).

SG - SingaporeEngineeringFull-timeSenior · 5+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
Salary not listedView Job
Clera logo
Research Engineer
Clera
11d ago

Apply for the Research Engineer role at Clera in Zürich. Build ML pipelines and post-training infrastructure. Competitive pay and visa sponsorship.

ZürichEngineeringFull-timeOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
S
GPU Kernel Engineer
Sciforium
11d ago

Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.

San Francisco, CAEngineeringFull-timeSenior · 5+ yrsOpen · checked 7d ago
PythonPyTorchJAX
$190K – $250KView Job
T
Machine Learning Engineer, Applied
tracelabs
11d ago

Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.

United StatesEngineeringFull-timeMid · 2+ yrsRemoteOpen · checked 7d ago
PythonPyTorchJAX
$185K – $245KView Job
Thinking Machines logo
RL Systems - Infra
Thinking Machines
11d ago

Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.

San FranciscoResearch Infrastructure (ML Infrastructure and Training Stack)Full-timeSeniorOpen · checked 7d ago
PyTorchJAXKubernetes
$425K - $475K USDView Job
P
Member of Technical Staff, AI Research
psi
11d ago

Apply for the AI Research Member of Technical Staff role at psi in Boston. Build AI agents and training systems to discover physics at scale.

BostonTechnical StaffFull-timeSeniorRemoteOpen · checked 7d ago
PyTorchJAXReinforcement Learning
Salary not listedView Job
P
Member of Technical Staff, Applied AI
psi
11d ago

Apply for Member of Technical Staff, Applied AI at psi in Boston. Build AI systems to discover new physics at scale. Check requirements and apply.

BostonTechnical StaffFull-timeSeniorRemoteOpen · checked 7d ago
PyTorchJAXMachine Learning
Salary not listedView Job
Planet logo
Senior Software Engineer
Planet
12d ago

Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.

Ljubljana, SloveniaAccessFull-timeSenior · 6+ yrsOpen · checked 7d ago
PythonJAXLLMs
€57.700 — €72.100 EURView Job
W
2027 Summer Intern, MS/PhD, Research, Perception Foundation Models
Waymo
12d ago

Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
T
Member of Technical Staff - ML Infra
trajectory
12d ago

Build ML infrastructure for AI systems at Trajectory in San Francisco. Focus on training, inference, and GPU kernels. Apply on the official careers page.

San FranciscoMember of Technical StaffFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
Lyft logo
PhD Machine Learning Software Engineer Intern (Summer 2027)
Lyft
12d ago

Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.

San Francisco, CAEarly TalentInternshipInternOpen · checked 7d ago
PythonPyTorchTensorFlow
$65-$68View Job
Mistral AI logo
LLM Scientist, AI for Science
Mistral AI
12d ago

Apply for the LLM Scientist, AI for Science role at Mistral in Paris. Develop LLMs and deep learning frameworks for biology, chemistry, and physics.

ParisScienceFull-timeSeniorRemoteOpen · checked 7d ago
PyTorchTensorFlowJAX
Salary not listedView Job
F
Senior Machine Learning Scientist
Flagship Pioneering, Inc.
13d ago

Apply as a Senior Machine Learning Scientist at Flagship Pioneering in Cambridge, UK. Build models and AI agents for somatic genomics.

Cambridge, UKArtificial IntelligenceFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
S
Senior HPC & GPU Infrastructure Engineer
Sciforium
13d ago

Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.

San Francisco, CAEngineeringFull-timeSenior · 5+ yrsOpen · checked 7d ago
PythonPyTorchJAX
$150K – $220KView Job