◆finddatasciencejobs.
Add as a preferred source on GooglePost a job
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.

Who’s hiring for JAX

B
BJAK80
W
Waymo39
MA
Mistral AI15
A
Anthropic11
JC
JPMorgan Chase9
R
Roche9
SI

Machine Learning Engineering Internship: Summer 2027

NEW
Susquehanna International Group·Hong Kong, Hong Kong·On-site
Junior·ML Engineer
PythonPyTorchTensorFlowBig data
2d ago→
A

Senior Software Engineer, Machine Learning Platform

NEW
Airwallex·SG - Singapore·Hybrid
Senior·ML Engineer
PythonMachine learningKubernetesDocker
2d ago→
R

Senior Machine Learning Research Scientist, Structure and Simulation (AI for Drug Discovery)

NEW
Roche·New York City; South San Francisco·On-site
Senior·Research Scientist·$141k – $262k
Machine learningDeep learningFoundation modelsSimulation
2d ago→
W

2027 Summer Intern, PhD, Machine Learning, Planning/Prediction

NEW
Waymo·Mountain View, CA, USA·On-site
Junior·Research Scientist·$177k
PythonC++Machine learningPlanning
2d ago→
T

Machine Learning Engineer, Applied

Trace·United States, Remote, NYC·Hybrid
Mid·ML Engineer·$185k – $245k
PythonMachine learningRoboticsData engineering
3d ago→
MA

LLM Scientist, AI for Science

Mistral AI·Paris·Hybrid
Senior·Research Scientist
LLMsFine-tuningDistributed trainingPython
4d ago→
I

Senior Bioinformatics Scientist – AI/ML for Protein Engineering

Illumina·US - California - San Diego·On-site
Senior·Research Scientist·$129k – $194k
BioinformaticsProtein engineeringMachine learningDeep learning
4d ago→
Z

Engineering Manager, ML Performance Optimization

Zoox·Foster City, CA·On-site
Senior·ML Engineer
Machine learningPerformance profilingGpu optimizationC++
4d ago→
L

PhD Machine Learning Software Engineer Intern (Summer 2027)

Lyft·San Francisco, CA·On-site
Junior·ML Engineer·$135k – $141k
PythonTensorFlowPyTorchSpark
4d ago→
W

AI Research Scientist, Scientific ML

Westerndigital·Singapore, Singapore·On-site
Senior·Research Scientist
Physics-informed neural networksLoss function designDigital twinNeural network training
4d ago→
A

Spring/Summer 2027 PhD AI Agentic/ML System Co-op

AMD·San Jose, California·On-site
Mid·Research Scientist
PythonMachine learningDeep learningReinforcement learning
5d ago→
R

Machine Learning Scientist/Senior Machine Learning Scientist - Synthesis Planning and Optimization, AI for Drug Discovery

Roche·San Francisco; New York City·On-site
Senior·Research Scientist·$148k – $274k
Machine learningDeep learningDrug discoverySynthesis planning
5d ago→
IO

Machine Learning Engineer — Reinforcement Learning

Institute of Foundation Models·Sunnyvale, CA·On-site
Senior·ML Engineer
PythonPyTorchTensorFlowReinforcement learning
5d ago→
IO

Machine Learning Engineer — Pre-training (LLM)

Institute of Foundation Models·Sunnyvale, CA·On-site
Senior·ML Engineer
PythonPyTorchTensorFlowLLMs
5d ago→
Z

Staff Machine Learning Engineer

Zscaler·Bellevue, Washington, USA; Dallas, Texas, USA; New York City, New York, USA; Santa Clara, California, USA·On-site
Staff·ML Engineer·$152k – $190k
PythonTensorFlowPyTorchAWS
5d ago→
FP

Senior Machine Learning Scientist

Flagship Pioneering, Inc·Cambridge, UK·On-site
Senior·Research Scientist
PythonMachine learningDeep learningGenomics
5d ago→
A

Machine Learning Engineer Intern

ATOMS·San Francisco, CA·On-site
Junior·ML Engineer·$146k
PythonMachine learningDeep learningComputer vision
5d ago→
AI

Senior Computer Vision Engineer

Anduril Industries·Irvine, California, United States·On-site
Senior·AI Engineer
Computer visionDeep learningSensor fusionReal-time processing
5d ago→
PA

Principal Applied AI Researcher (Prisma AIRS)

Palo Alto Networks·Santa Clara, United States of America·On-site
Staff·Research Scientist·$163k – $264k
Machine learningDeep learningPythonPyTorch
1w ago→
R

Senior Machine Learning Scientist, AI for Biology & Translation (AIBT)

Roche·South San Francisco, California, United States of America·On-site
Senior·Research Scientist·$1664k – $274k
Machine learningDeep learningBioinformaticsGenomics
1w ago→
M

Data Scientist

Millennium·Hong Kong, Hong Kong·On-site
Senior·Data Scientist
PythonRSQLC++
1w ago→
A

AI/ML Solutions Engineer

AMD·Santa Clara, California·On-site
Mid·AI Engineer
PythonC++Deep learningGPU
1w ago→
GM

2027 Summer Intern – AI/ML Engineer, Autonomous Vehicles: Simulation (PhD)

General Motors·Sunnyvale, California, United States of America·On-site
Junior·ML Engineer·$148k – $175k
PythonPyTorchTensorFlowComputer vision
1w ago→
W

Staff Machine Learning Engineer, Maneuvering Tech

Waymo·Mountain View, CA, USA; San Francisco, CA, USA; Pittsburgh, PA, USA·On-site
Staff·ML Engineer
PythonTensorFlowPyTorchComputer vision
1w ago→
W

Senior Machine Learning Engineer, Maneuvering Tech

Waymo·Mountain View, CA, USA; San Francisco, CA, USA; Pittsburgh, PA, USA·On-site
Senior·ML Engineer·$213k – $263k
PythonTensorFlowPyTorchReinforcement learning
1w ago→
W

Machine Learning Engineer, Maneuvering Tech

Waymo·Mountain View, CA, USA; San Francisco, CA, USA; Pittsburgh, PA, USA·On-site
Senior·ML Engineer·$175k – $215k
PythonTensorFlowPyTorchComputer vision
1w ago→
M

Associate Scientist, Post Doc Fellow- AI/ML & Computational Biology for Antigen Design

MSD·USA - Pennsylvania - West Point·On-site
Senior·Research Scientist·$82k – $92k
AIMachine learningComputational biologyProtein design
1w ago→
A

Sr. Associate - Machine Learning Engineer

Amgen·India - Hyderabad·On-site
Senior·ML Engineer
PythonPyTorchTensorFlowLLMs
1w ago→
W

2027 Summer Intern, PhD, Planner Machine Learning

Waymo·San Francisco, California·On-site
Junior·Research Scientist·$177k
PythonC++Machine learningPlanning
1w ago→
A

Research Engineer / Research Scientist, RL Frontiers

Anthropic·San Francisco, CA | New York City, NY | Seattle, WA·On-site
Senior·Research Scientist·$500k – $850k
Reinforcement learningDeep learningPythonPyTorch
1w ago→
C

ML Algorithm Mapping and Performance Engineer, Core ML

Cerebras·Sunnyvale, CA·On-site
ML Engineer
Core mlInferencePerformance engineeringAlgorithm mapping
1w ago→
C

ML Runtime and Kernel Engineer - Core ML

Cerebras·Sunnyvale, CA, Toronto, CAN·On-site
Senior·ML Engineer
Machine learningRuntimeKernelInference
1w ago→
R

Machine Learning Scientist/Senior Machine Learning Scientist - Agents for Applied Small Molecule Drug Design, AI for Drug Discovery

Roche·San Francisco; New York City·On-site
Senior·Research Scientist·$148k – $274k
Machine learningDeep learningPythonPyTorch
1w ago→
N

Quantitative Analyst, Front Office (Calypso)

Nasdaq·Paris·On-site
Mid·Analytics Engineer
PythonC++JavaRisk modeling
1w ago→
A

AI Research Scientist

Autodesk·Toronto, ON, CAN·On-site
Senior·Research Scientist·$173k – $253k
Deep learningComputer visionNatural language processingMultimodal learning
1w ago→
W

2027 Summer Intern, PhD, Machine Learning Research, Planning/Prediction

Waymo·Mountain View, CA, USA·On-site
Junior·Research Scientist·$177k
PythonPyTorchTensorFlowReinforcement learning
1w ago→
T

Lead Data Scientist - Recommendations (applied ML, Reinforcement Learning, Contextual Bandit Design)

Target·1000 Nicollet Mall, Minneapolis,MN 55403-2542·On-site
Senior·Data Scientist·$132k – $238k
PythonReinforcement learningContextual banditsRecommendation systems
2w ago→
H

Senior Data Engineer, Professional Services

Hitachi·(DEAI HV) US Remote VA - Washington DC Metro·Hybrid
Senior·Data Engineer·$160k – $205k
PythonSQLSparkAirflow
2w ago→
A

Principal Engineer, Efficient GenAI

AMD·San Jose, California·On-site
Staff·AI Engineer
Generative AIHardware accelerationGPUSilicon design
2w ago→
MA

Applied Science - Research Engineer (Robotics)

Mistral AI·Singapore·Hybrid
Senior·Research Scientist
RoboticsComputer visionDeep learningReinforcement learning
2w ago→
W

2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning

Waymo·Mountain View, CA, USA·On-site
Junior·ML Engineer·$146k
PythonC++Machine learningComputer vision
2w ago→
W

2027 Summer Intern, MS/PhD, Perception, Machine Learning

Waymo·Mountain View, CA, USA·On-site
Junior·Research Scientist·$146k
PythonC++Deep learningComputer vision
2w ago→
AI

Research Scientist, Climate Modeling

Allen Institute·Seattle, WA·On-site
Senior·Research Scientist·$167k – $261k
Machine learningDeep learningPhysics-informed neural networksClimate modeling
2w ago→
AI

Research Engineer, Climate Modeling

Allen Institute·Seattle, WA·On-site
Senior·Research Scientist·$147k – $220k
PythonMachine learningDeep learningClimate modeling
2w ago→
FP

Senior Director / Vice President, Machine Learning (Generative Chemistry)

Flagship Pioneering, Inc·Cambridge, MA USA·On-site
Senior·Research Scientist·$279k – $413k
Machine learningDeep learningGenerative AICheminformatics
2w ago→
B

Applied AI Engineer

BJAK·China·On-site
Mid·AI Engineer
Machine learningNatural language processingPrompt engineeringModel fine-tuning
2w ago→
L

Enterprise Data Scientist

Lseg·SGP-Singapore-1 Raffles Quay; HKG-Hong Kong-ICBC Tower; IND-Mumbai-One World Center·On-site
Senior·Data Scientist
AIMachine learningGenerative AINatural language processing
2w ago→
CD

Research Engineer - ML Infrastructure

Chai Discovery·San Francisco office·On-site
Mid·MLOps
PythonPyTorchKubernetesDocker
2w ago→
CD

Research Engineer - Auto Research

Chai Discovery·San Francisco office·On-site
Mid·Research Scientist
PythonPyTorchTensorFlowGraph neural networks
2w ago→
SI

Machine Learning Engineering Internship

Susquehanna International Group·Bala Cynwyd (Philadelphia Area), Pennsylvania·On-site
Junior·ML Engineer
PythonPyTorchTensorFlowSQL
2w ago→

JAX jobs by city

San Francisco (88)New York (48)Mountain View (40)London (38)Seattle (22)Paris (16)

Skills often asked for alongside JAX

PyTorch (435)AI (403)Python (391)Machine Learning (384)Deep Learning (240)LLMs (213)TensorFlow (198)Software Engineering (190)

Other skills in demand

SQLData ScienceStatisticsData PipelinesAWSData EngineeringData VisualizationData QualityAzureCloud ComputingSparkAgentic AI
All skills →

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.