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.

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→

Research Engineer (LLM Performance), London

Isomorphic Labs·London·On-site

Senior·Research Scientist

Python PyTorch LLMs Performance optimization

1mo ago→

Machine Learning Engineer I/II, Applied AI

Lila Sciences·Cambridge, MA USA; San Francisco, CA USA·On-site

Mid·ML Engineer·$116k – $170k

Python PyTorch TensorFlow Model training

1mo ago→

Sr. Software Engineer, AI / ML Inference Platform

Dialpad·Buenos Aires, Argentina·On-site

Senior·MLOps

Python MLOps Distributed systems Real time inference

1mo ago→

Senior AI Engineer

Finaira·Cairo, Cairo Governorate, Egypt·On-site

Senior·AI Engineer

Generative AI LLMs Reinforcement learning Anomaly detection

1mo ago→

ML Engineer

Monarch·Emeryville, California·On-site

Mid·MLOps

MLOps Data pipelines Version control Model evaluation

1mo ago→

VP AI Research/Applied AI - Deep Learning & Time Series

Campus·New York, NY, United States·On-site

·$150k – $300k

Machine learning AI Deep learning Time series

1mo ago→

Research Engineer, Forge

Mistral AI·Paris, Palo Alto, New York, Zurich, Amsterdam, Linz, Munich, Singapore, London·Hybrid

AI Machine learning Distributed training Python

1mo ago→

AI Scientist - Physics Models

Mistral AI·Paris, Amsterdam, Linz, Munich, London, Berlin·On-site

AI Machine learning Data pipelines Python

1mo ago→

Tech Lead Manager ML Optimization

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

·$298k – $378k

ML infrastructure Machine learning AI Data engineering

1mo ago→

Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference

Snapchat·Los Angeles, California; Seattle, Washington; Palo Alto, California; New York, New York; Bellevue, Washington·On-site

Staff·ML Engineer·$229k – $343k

Python PyTorch Diffusion models Generative modeling

1mo ago→

Embedded AI Security Research Engineer (AI MSc / PhD / Postdoc) (m/f/d)

Nxp·Leuven; Gratkorn·On-site

Mid·Research Scientist

Adversarial ai defense Privacy preserving ml Confidential computing Ai model protection

1mo ago→

AI/ML Engineer

Caci International·International - Germany·On-site

·$126k – $277k

AI Machine learning GCP MLOps

1mo ago→

AI Research Scientist, New Grad – Agents & Reinforcement Learning

Snowflake·US-WA-Bellevue·On-site

Junior·Research Scientist·$176k – $230k

Reinforcement learning Agentic systems Experimental ai prototyping Distributed systems

1mo ago→

Senior AI/ML Engineer (F/M/D)

Animore·Munich, Bavaria, Germany·On-site

Senior·AI Engineer

Foundation models Computer vision Natural language processing Robotics

1mo ago→

Technical Lead Manager, Prediction, ML Evaluation

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

·$251k – $310k

Machine learning Neural networks Model evaluation TensorFlow

1mo ago→

Staff Tech Lead, Machine Learning Engineer, Perception

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

·$251k – $310k

Deep learning Machine learning Computer vision Python

1mo ago→

Staff Research Scientist, Foundation Models Recipes

Waymo·Mountain View, CA, US; San Francisco, CA, US; Kirkland, WA, US·On-site

·$251k – $310k

AI Machine learning Reinforcement learning Bayesian statistics

1mo ago→

Staff ML Engineer, Perception Research

Waymo·Mountain View, CA, USA ; New York City, NY, USA ; Kirkland, WA, USA ; San Francisco, CA, USA·On-site

·$251k – $310k

AI Machine learning Reinforcement learning Bayesian statistics

1mo ago→

Staff Machine Learning Engineer – VLM/LLM

Waymo·Mountain View, CA, USA; San Francisco, CA, USA; New York City, NY, USA·On-site

·$238k – $302k

AI Machine learning Reinforcement learning Bayesian statistics

1mo ago→

Sr Staff Tech Lead, Machine Learning Engineer, Perception

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

·$298k – $368k

Machine learning AI Computer vision Python

1mo ago→

Senior Staff ML Engineer, Driver Understanding and Evaluation

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

·$281k – $356k

Machine learning Fine-tuning Generative AI Deep learning

1mo ago→

Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & Optimization

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

·$298k – $368k

LLMs Software engineering Machine learning Deep learning

1mo ago→

Senior/Staff Machine Learning Engineer, Infrastructure

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

·$251k – $310k

ML infrastructure JAX TensorFlow Reinforcement learning

1mo ago→

Senior Research Scientist, AI Foundations Recipes

Waymo·Oxford, UK ; London, UK·On-site

·$123k – $129k

AI Machine learning Reinforcement learning Bayesian statistics

1mo ago→

Senior Machine Learning Engineer, Runtime and Serving

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

·$213k – $263k

Machine learning Deep learning JAX Software engineering

1mo ago→

Senior Machine Learning Engineer, Multimodal Perception

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

·$213k – $263k

Deep learning PyTorch JAX AI

1mo ago→

Senior Machine Learning Engineer, Computer Vision/VLM

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

·$213k – $263k

Computer vision Deep learning Generative AI AI

1mo ago→

Perception Machine Learning Engineer - Continuous Learning

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

·$175k – $215k

Machine learning Data infrastructure Data pipelines Computer vision

1mo ago→

Perception Machine Learning Engineer

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

·$170k – $216k

Machine learning Fine-tuning Model evaluation Deep learning

1mo ago→

Machine Learning Engineer, Prediction & Planning

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

·$175k – $215k

Machine learning Neural networks Reinforcement learning Deep learning

1mo ago→

Machine Learning Engineer / Applied Scientist, Prediction & Planning

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

·$175k – $215k

Machine learning Neural networks Generative AI Reinforcement learning

1mo ago→

Senior / Principal ML Scientist, Foundation Models for Life Sciences

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

Staff·ML Engineer·$268k – $384k

Distributed training pipelines Foundation models Antibody biologics design Multimodal learning

1mo ago→

Senior / Engineer II, AI Lab Research Engineer

Lila Sciences·Cambridge, MA USA; San Francisco, CA USA·On-site

Senior·AI Engineer·$148k – $240k

Reinforcement learning Autonomous lab systems Scientific ai Molecular modeling

1mo ago→

Scientist II / Senior ML Scientist, Data-Efficient Learning for Drug Discovery

Lila Sciences·Cambridge, MA USA; London, UK; San Francisco, CA USA·On-site

Senior·Research Scientist·$228k – $358k

Active learning Meta learning Uncertainty estimation Multimodal modeling

1mo ago→

Principal AI Researcher

Workday·USA, CA, Pleasanton; USA, GA, Atlanta; Canada, BC, Vancouver·On-site

Staff·Research Scientist

Machine learning research Ai platform architecture Natural language processing Data modeling

1mo ago→

Senior AI Engineer

Centific·Redmond, Washington·On-site

·$160k – $220k

AI LLMs RAG Generative AI

1mo ago→

Principal Research Scientist – Vision AI, Simulation & Physical AI

Centific·Remote Work( USA)·Remote

·$200k – $300k

AI LLMs RAG Generative AI

1mo ago→

ML Infra Engineer, Modeling

Physical Intelligence·San Francisco·On-site

AI JAX Software engineering ML infrastructure

1mo ago→

Senior Research Engineer, Olmo + Molmo

Allen Institute·Seattle, WA·On-site

·$174k – $261k

Deep learning Python AI Generative AI

1mo ago→

page 3 of 9 · 450 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