updated october 2026

JAX Jobs in Data Science (October 2026)

447 open data science, ML and AI roles ask for JAX as of October 2026, across 172 hiring companies — 102 of them remote-friendly. BJAK is hiring the most, with 80 openings. San Francisco is the busiest market. Listings are updated daily, and every role links straight to the employer's official application.

Staff Tech Lead Manager, Machine Learning, Simulator Evaluation

Waymo·Mountain View, California, United States; San Francisco, California, United States. ·On-site

Staff·ML Engineer·$251k – $310k

Python C++TensorFlow PyTorch

3w ago→

2027 Summer Intern, PhD, Machine Learning, Computer Vision

Waymo·Mountain View, California, United States·On-site

Junior·ML Engineer·$177k

Computer vision Deep learning Autonomous vehicle perception Reinforcement learning

3w ago→

Executive Director (Senior Lead Securities Python Quantitative Developer )

Wf·CHARLOTTE, NC·On-site

Senior·Data Engineer

Python Quantitative development Risk analytics Frtb compliance

3w ago→

Senior Deep Learning Engineer, Cosmo 3D Spatial

NVIDIA·US, CA, Santa Clara·On-site

Senior·AI Engineer

3d geometry processing Spatial reasoning models Metric learning Cross view correspondence

3w ago→

2027 PhD Graduate - Artificial Intelligence and Complex Systems

Johns Hopkins APL·Laurel, Maryland·On-site

AI Machine learning Agentic AI Statistics

3w ago→

AI and Computer Vision Engineer

The Exploration Company·Munich, Germany·On-site

AI Computer vision Deep learning Distributed training

3w ago→

Senior Machine Learning Engineer, Ads Response Prediction

Instacart·Canada - Remote (ON, AB, BC, or NS Only)·Hybrid

Senior·ML Engineer·$180k – $190k

Feature engineering Model deployment Recommendation systems A/B testing

3w ago→

Applied Scientist – User Intelligence

InMobi·San Mateo, CA·On-site

·$182k – $219k

AI Machine learning Data science Deep learning

3w ago→

Principal Applied Scientist

Oracle·United States·On-site

·$126k – $264k

Data intelligence AI Machine learning LLMs

3w ago→

Staff Machine Learning Engineer

Phantom·Remote·Remote

Machine learning Data science Reinforcement learning LLMs

3w ago→

Senior Applied ML Engineer

CoreWeave·Bellevue, WA / Sunnyvale, CA ·On-site

Senior·ML Engineer·$182k – $242k

Distributed systems architecture Large-scale ml training Gpu/accelerated computing Cloud infrastructure optimization

3w ago→

Senior Localization Engineer / Research Engineer, Robotics AI (Visual Navigation)

Grab·Singapore, Singapore·On-site

AI Deep learning Data quality Computer vision

4w ago→

Quantitative Trading & Research - Quantitative Developer - Associate or Vice President

JPMorgan Chase·LONDON, LONDON, United Kingdom·On-site

Mid·AI Engineer

Quantitative research Low latency engineering ML infrastructure Market microstructure

4w ago→

Quantitative Trading & Research - AI/ML Quantitative Researcher - Associate or Vice President

JPMorgan Chase·LONDON, LONDON, United Kingdom·On-site

Mid·AI Engineer

Transformer models Time series forecasting Large scale data processing Market microstructure

4w ago→

AI Engineering Team Lead

and at Syngenta·Beijing, Beijing, China·On-site

Senior·AI Engineer

Python Distributed computing Genomics pipelines MLOps

4w ago→

Machine Learning Engineering Technical Leader - CX AI

Cisco·San Jose, California, US·On-site

Senior·ML Engineer·$257k – $333k

Machine learning Deep learning LLMs Inference

4w ago→

Senior Software Engineer, Machine Learning (Safety)

Discord·San Francisco Bay Area·On-site

Senior·ML Engineer·$220k – $248k

Python TensorFlow PyTorch Natural language processing

4w ago→

Staff ML Engineer - Embodied AI Onboard Autonomy

General Motors·Sunnyvale, California, United States of America·On-site

·$180k – $280k

AI Machine learning Neural networks LLMs

1mo ago→

Senior Machine Learning Engineer, Vehicle Interaction & Behavior Policy ML

Waymo·Mountain View, California, USA·On-site

·$213k – $263k

Machine learning AI PyTorch JAX

1mo ago→

Senior Machine Learning Engineer, Predictive Planner, Systems Arch

Waymo·Mountain View, CA, USA; San Francisco, CA, USA·On-site

·$213k – $263k

Machine learning Neural networks Data infrastructure Deep learning

1mo ago→

2027 Summer Intern, PhD, Machine Learning, Computer Vision

Waymo·Mountain View, California·On-site

·$177k

Software engineering Deep learning Neural networks Python

1mo ago→

ML Scientist, Nucleic Acid Design

Lila Sciences·San Francisco, CA USA·On-site

Mid·Research Scientist·$176k – $304k

Sequence modeling Regulatory genomics 3d utr 5d utr optimization Cds optimization

1mo ago→

ML Scientist, Foundation Models for Life Sciences

Lila Sciences·San Francisco, CA USA·On-site

Mid·Research Scientist·$176k – $304k

Protein structure prediction Cofolding models Generative ai for biology Deep learning

1mo ago→

AI Research Engineer - Datadog AI Research (DAIR)

Datadog·Paris, France·On-site

Senior·AI Engineer

Machine learning prototyping Time series Cloud observability Model evaluation frameworks

1mo ago→

Staff Machine Learning Engineer

ATOMS·San Francisco, CA·On-site

Staff·ML Engineer·$273k – $345k

Reinforcement learning Robotics control systems Hardware-software integration Real-time sensor data processing

1mo ago→

Senior Machine Learning Engineer

ATOMS·San Francisco, CA·On-site

Senior·ML Engineer·$208k – $264k

Reinforcement learning Robotics control systems Hardware-software integration Real world ai deployment

1mo ago→

AI Engineer (Managed Services)

AvePoint·Singapore·On-site

Senior·AI Engineer

LLMs Rag systems Prompt engineering Agent orchestration

1mo ago→

Quantitative Trading & Research, Quantitative Developer - Associate

JPMorgan Chase·Singapore·On-site

Mid·AI Engineer

Low-latency systems Quantitative research ML infrastructure Market microstructure

1mo ago→

Quantitative Trading & Research - AI Scientist, Quantitative Research - Associate

JPMorgan Chase·Singapore·On-site

Senior·Research Scientist

Transformer-models Foundation-model-pre-training Time-series-forecasting Market-microstructure-analysis

1mo ago→

AI Research Scientist - Safety

Bosch Group·Sunnyvale, CA, United States·On-site

Senior·Research Scientist·$165k – $185k

Ai safety Model alignment Generative modeling Simulation frameworks

1mo ago→

Senior Data Scientist - Protein Structure ML models

Amgen·India - Hyderabad·On-site

Senior·Data Scientist

Machine learning Deep learning Protein structure prediction Antibody modeling

1mo ago→

AI Researcher

Hudson River Trading·London, United Kingdom; New York, NY, United States·On-site

·$200k – $300k

AI Deep learning LLMs PyTorch

1mo ago→

AI Research Engineer, Pre-Training

Hudson River Trading·London, United Kingdom; New York, NY, United States·On-site

·$250k – $300k

AI PyTorch JAX Deep learning

1mo ago→

AI Research Engineer, Inference

Hudson River Trading·London, United Kingdom; New York, NY, United States·On-site

·$250k – $300k

AI PyTorch JAX Deep learning

1mo ago→

Machine Learning Engineer (II-III), Space Edge Deployment

True Anomaly·Denver, Colorado or Long Beach, California·On-site

·$125k – $220k

Machine learning AI Anomaly detection Feature engineering

1mo ago→

Machine Learning Engineer (II-III), Space Agentic AI

True Anomaly·Denver, CO or Long Beach, CA·On-site

·$125k – $220k

Machine learning AI Anomaly detection Feature engineering

1mo ago→

Senior AI Engineer

Suki·Redwood City, CA·On-site

·$468k – $250k

AI GCP Kubernetes Go

1mo ago→

Staff+ Software Engineer, ML Inference Path

Anthropic·San Francisco, CA·On-site

Staff·MLOps·$320k – $485k

MLOps Model-inference Safety-classifiers Production-deployment

1mo ago→

Quantitative Trading & Research - Quantitative Developer - Associate

JPMorgan Chase·Central and Western, Hong Kong Island, Hong Kong·On-site

Mid·AI Engineer

Low-latency systems Quantitative research ML infrastructure Market microstructure

1mo ago→

Quantitative Trading & Research - AI Scientist, Quantitative Research - Associate/ Vice President

JPMorgan Chase·Central and Western, Hong Kong Island, Hong Kong·On-site

Senior·AI Engineer

Transformer architectures LLMs Time series forecasting Market microstructure analysis

1mo ago→

AI Scientist (Model Building & Training)

Lilly·US, San Francisco CA·On-site

Senior·AI Engineer·$168k – $268k

Deep learning Drug discovery modeling Biomedical data pipelines Hyperparameter optimization

1mo ago→

ML Infrastructure Engineer

xAI· Palo Alto, CA·On-site

·$180k – $440k

AI ML infrastructure Recommendation systems Data pipelines

1mo ago→

Staff Machine Learning Engineer, Mapping

Waymo·Mountain View, CA, USA·On-site

·$251k – $310k

Machine learning Generative AI AI Computer vision

1mo ago→

Senior Software Engineer, ML Evaluation Infra and Efficiency

Waymo·Mountain View, California, United States·On-site

·$238k – $302k

ML infrastructure JAX TensorFlow Distributed systems

1mo ago→

Research Scientist, Map Scalability

Waymo·Mountain View, CA, U.S.A ; San Francisco, CA, U.S.A ; New York City, NY, U.S.A·On-site

·$213k – $263k

Machine learning Reinforcement learning Bayesian statistics AI

1mo ago→

Machine Learning Research Engineer

Optiver·New York, New York·On-site

Mid·ML Engineer·$200k

Deep learning Price forecasting Scalable pipelines Framework customization

1mo ago→

Principal Machine Learning Engineer, Foundation Models, AI for Drug Discovery

Roche·New York City, New York, United States of America; South San Francisco, California, United States of America·On-site

Staff·ML Engineer·$193k – $358k

Foundation models Drug discovery ai LLMs Distributed computing

1mo ago→

Quantitative Trading & Research - Quantitative Developer Systematic Trading - Associate

JPMorgan Chase·New York, NY, United States·On-site

Mid·AI Engineer

ML infrastructure Low latency engineering Market microstructure analysis Quantitative research

1mo ago→

Quantitative Trading & Research - AI Scientist - Associate

JPMorgan Chase·New York, NY, United States·On-site

Mid·AI Engineer

Transformer architectures Large language model pretraining Time-series forecasting Quantitative finance modeling

1mo ago→

Federal Data Engineer

Hitachi·(DEAI HV) US Remote DC·Hybrid

Senior·Data Engineer·$160k – $205k

Python SQL Spark AWS

1mo ago→

page 2 of 10 · 452 jobs

← Previous Next →

Methodology: live data science, ML and AI listings tagged JAX. A job is tagged when its description names the skill, or when our enrichment step identifies it from the role. Counts are live and change daily as roles open and close.

Who’s hiring for JAX

JAX jobs by city

Skills often asked for alongside JAX

Other skills in demand