Latest AI JobsBrowse Jobs

Jax Jobs

191 open roles, updated daily

Filters1
191 jobs foundClear all
A
Machine Learning Engineer Intern
ATOMS Careers page
13d ago

Join Atoms as a Machine Learning Engineer Intern in San Francisco. Build physical AI and autonomous transport platforms. $70/hr.

San Francisco, CATransport - EngineeringInternshipInternOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
W
Senior Software Engineer, Vehicle Interaction
Waymo
13d ago

Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.

Mountain View, CA, USAPlanner (7LU)Full-timeSenior · 5+ yrsOpen · checked 7d ago
PyTorchJAXC++
$213K — $263K USDView Job
Anduril Industries logo
Senior Computer Vision Engineer
Anduril Industries
13d ago

Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.

Irvine, California, United StatesAFS : Counter Intrusion Engineering : Sentry Software EngineeringFull-timeSenior · 6+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$220K — $292K USDView Job
W
2027 Summer Intern, PhD, Research, World Action Model
Waymo
16d ago

Research and implement world action models for autonomous driving at Waymo. Ideal for a Ph.D. candidate with a strong ML background.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
A
Machine Learning Engineer (5-8 yrs)
Advanced Space
16d ago

Develop ML solutions for lunar and Mars missions. 5-8 years experience required in autonomy, navigation, and mission planning.

Westminster, COMachine LearningFull-timeMid · 5+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$124K -$171KView Job
M
Research Engineer, Post-training
medraai
17d ago

Apply for the Research Engineer, Post-training role at medraai in San Francisco. Build Physical AI systems for biopharma partners.

San FranciscoEngineeringFull-timeOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
W
2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement Foundations
Waymo
17d ago

Join Waymo as a 2027 Summer Intern in Machine Learning for Driver Refinement Foundations. Work on LLMs, generative models, and autonomous driving tech.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$70 — $70 USDView Job
NormalComputing logo
Research Engineer, Agentic EDA
NormalComputing
19d ago

Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.

New York CityEngineeringFull-timeSeniorOpen · checked 7d ago
PythonPyTorchJAX
$200K – $400KView Job
Anthropic logo
Research Engineer / Research Scientist, RL Frontiers
Anthropic
19d ago

Join Anthropic's RL Scaling team to build next-gen algorithms and scale reinforcement learning for Claude. Annual salary $500,000 — $850,000 USD.

San Francisco, CA | New York City, NY | Seattle, WAAI Research & EngineeringFull-timeOpen · checked 7d ago
PythonPyTorchJAX
$500K — $850K USDView Job
U
CAD Automation Researcher
universalagi
19d ago

Apply for the CAD Automation Researcher role at universalagi in San Francisco. Build AI foundation models for CAD geometry and physics simulation.

San FranciscoTechnical StaffFull-timeSenior · 2+ yrsOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
W
2027 Summer Intern, PhD, Planner Machine Learning
Waymo
19d ago

Join Waymo as a PhD Planner Machine Learning Intern in San Francisco. Research deep learning architectures for autonomous driving. Apply now.

San Francisco, CaliforniaPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Machine Learning Research, Planning/Prediction
Waymo
20d ago

Join Waymo as a PhD Machine Learning Research Intern in Mountain View. Research generative models for autonomous driving planning and prediction.

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

Apply for the Machine Learning Platform Engineer role at bjakcareer in Seoul, Korea. Build ML infrastructure, platforms, and production AI systems.

Seoul, KoreaActAI EngineeringFull-timeRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Member of Technical Staff, Machine Learning
bjakcareer
21d ago

Apply for the Member of Technical Staff, Machine Learning role at bjakcareer in Poland. Build production ML systems, Python, PyTorch, JAX.

PolandActAI EngineeringFull-timeRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Senior Machine Learning Engineer
bjakcareer
21d ago

Join ACTAI as a Senior Machine Learning Engineer in Spain. Build core ML systems and own production-ready AI applications end-to-end.

SpainActAI EngineeringFull-timeSeniorRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Senior Machine Learning Engineer
bjakcareer
21d ago

Apply for the Senior Machine Learning Engineer role at ACTAI in Poland. Build core ML systems for proactive AI products.

PolandActAI EngineeringFull-timeSeniorRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Machine Learning Platform Engineer
bjakcareer
21d ago

Join ActAI in Germany as a Machine Learning Platform Engineer to build and operate scalable ML infrastructure, serving systems, and production pipelines.

GermanyActAI EngineeringFull-timeMidRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Member of Technical Staff, Machine Learning
bjakcareer
21d ago

Build core ML components for production systems at ACTAI in Zurich, Switzerland. Focus on fine-tuning, evaluation, and real-world task completion.

Zurich, SwitzerlandActAI EngineeringFull-timeRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Senior Machine Learning Engineer
bjakcareer
21d ago

Join bjakcareer in Zurich as a Senior Machine Learning Engineer. Own production ML systems and build proactive AI applications.

Zurich, SwitzerlandActAI EngineeringFull-timeSeniorRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Machine Learning Platform Engineer
bjakcareer
21d ago

Apply for the Machine Learning Platform Engineer role at bjakcareer in Portugal. Build scalable ML infrastructure and AI production systems.

PortugalActAI EngineeringFull-timeMidRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Machine Learning Platform Engineer
bjakcareer
21d ago

Apply for the Machine Learning Platform Engineer role at bjakcareer in Sweden. Build scalable AI infrastructure and production ML systems.

SwedenActAI EngineeringFull-timeMidRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Machine Learning Platform Engineer
bjakcareer
21d ago

Build ML infrastructure and serving systems for ActAI in the United Kingdom. Apply on the official careers page.

United KingdomActAI EngineeringFull-timeMidRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Machine Learning Platform Engineer
bjakcareer
21d ago

Build ML infrastructure and serving systems for ActAI in China. Full-time role focused on scalable AI platforms, distributed systems, and MLOps.

ChinaActAI EngineeringFull-timeMidRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job
bjakcareer logo
Senior Machine Learning Engineer
bjakcareer
21d ago

Join ACTAI in the United States as a Senior Machine Learning Engineer to build proactive AI systems. Apply on the official careers page.

United StatesActAI EngineeringFull-timeSeniorRemoteOpen · checked 7d ago
PythonPyTorchJAX
Salary not listedView Job