◆finddatasciencejobs.
Add as a preferred source on GooglePost a job
updated october 2026

JAX Jobs in Data Science (October 2026)

451 open data science, ML and AI roles ask for JAX as of October 2026, across 173 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
G
Graphcore10
JC
JPMorgan Chase9
A

ML Research Scientist (MLRS) - Representation Learning for Molecular AI

Achira·San Francisco Office, New York Office·On-site
Senior·Research Scientist·$165k – $259k
Representation learningGraph neural networksTransformer architecturesDistributed training
3mo ago→
A

ML Research Scientist (MLRS) - Generative AI

Achira·San Francisco Office, New York Office·Hybrid
·$165k – $259k
AIMachine learningGenerative AIBayesian statistics
3mo ago→
SA

Computer Vision AI & ML Engineer

Skild AI·Bengaluru, India·On-site
Mid·ML Engineer
Computer visionDeep learningConvolutional neural networksRobotics perception
3mo ago→
A

Physics Informed Machine Learning Scientist

ASML·San Diego, CA, USA·On-site
Mid·Research Scientist·$135k – $203k
Physics-informed-neural-networksLaser-plasma-interactionsOptical-systems-modelingControl-algorithm-development
3mo ago→
MA

AI Scientist

Mistral AI·Warsaw·On-site
AILLMsPythonRust
3mo ago→
A

Principal Machine Learning Engineer - Forecasting

Amgen·India - Hyderabad·On-site
Staff·ML Engineer
Time series forecastingUncertainty estimationLLMsAgentic AI
3mo ago→
Z

Machine Learning Engineer

Zillow·Remote-Germany·Remote
Senior·ML Engineer
PythonTensorFlowComputer visionDeep learning
3mo ago→
R

Machine Learning Scientist - Quantum Chemistry

Roche·Welwyn; Basel·On-site
Senior·Research Scientist
Quantum chemistry modelingSmall-molecule drug discoveryMachine learning for scientific applicationsHigh-performance computational methods
3mo ago→
A

Staff AI Engineer

Airwallex·US - San Francisco·On-site
Staff·AI Engineer·$182k – $310k
PythonMachine learningDeep learningDistributed systems
3mo ago→
R

Senior/Principal Machine Learning Scientist, Perturbation Biology, AI Biology & Translation (AIBT)

Roche·South San Francisco, California, United States of America·On-site
·$148k – $274k
AIMachine learningStatisticsPython
3mo ago→
A

Staff Research Scientist, Artificial Intelligence

Analogdevices·US, MA, Boston·On-site
·$172k – $237k
Machine learningAIDeep learningReinforcement learning
3mo ago→
M

AI Scientist

Millennium·New York, New York, United States of America·On-site
·$175k – $250k
AIData scienceMachine learningGenerative AI
3mo ago→
L

Senior ML Research Developer

Lawzero·Montreal·On-site
Senior·ML Engineer
Large-scale model trainingDistributed computing optimizationExperimental workflow designModel inference acceleration
3mo ago→
L

Senior ML Research Scientist

Lawzero·Montreal·On-site
Senior·Research Scientist
Machine learningDeep learningModel developmentAi safety
3mo ago→
G

AIML Engineer, AI for Science

GSK·UK - Kings Cross – London; Heidelberg; Switzerland - Zug·On-site
Mid·ML Engineer·$84k – $140k
Deep learningReinforcement learningHigh performance computingBiomedical data analysis
3mo ago→
GM

ML & Molecular Simulation Scientist

Genesis Molecular AI·San Mateo, CA·On-site
AIDeep learningMachine learningNeural networks
4mo ago→
IS

Senior AI/ML Research Engineer – Model development

Intuitive Surgical·Sunnyvale, CA, United States·On-site
Senior·AI Engineer
Foundation model finetuningMultimodal learningComputer visionEmbodied ai
4mo ago→
S

Research Scientist - Salesforce AI Research

Salesforce·California - Palo Alto; California - San Francisco·On-site
Senior·Research Scientist·$117k – $314k
Machine learningAi researchCrm ai systemsNatural language processing
4mo ago→
M

ML Engineer

Mach9·San Francisco·On-site
Mid·ML Engineer·$180k – $300k
Point cloud processing3d object detectionLidar data pipelinesComputer vision architectures
4mo ago→
D

Senior Machine Learning Engineering Manager, Ad Platforms

Disney·Seattle, WA, USA; Santa Monica, CA, USA·On-site
Senior·ML Engineer·$207k – $278k
Distributed systemsScalable ml pipelinesAd targeting algorithmsReal time data processing
4mo ago→
D

Manager, Machine Learning Engineering - Ad Platforms

Disney·Seattle, WA, USA; Santa Monica, CA, USA·On-site
Senior·ML Engineer·$172k – $230k
Distributed systemsScalable ml pipelinesAd targeting algorithmsReal time data processing
4mo ago→
SA

Computer Vision AI & ML Engineer

Skild AI·San Mateo, CA·On-site
Mid·AI Engineer
Computer visionDeep learningRobotics perceptionModel deployment
4mo ago→
S

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ·Canada·Hybrid
Senior·ML Engineer·$140k – $210k
LLMsQuantitative-modelingDeep-learning-frameworksDistributed-computing
4mo ago→
S

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ·Canada·Hybrid
Staff·ML Engineer·$172k – $258k
LLMsQuantitative-modelingDeep learningDistributed systems
4mo ago→
S

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ·United States·Hybrid
Senior·ML Engineer·$164k – $246k
LLMsQuantitative-modelingDeep learningReinforcement learning
4mo ago→
M*

Staff Data Scientist | NLP

Machinify *·Remote - US·Remote
Staff·Data Scientist·$180k – $230k
Natural language processingText classificationInformation extractionTransformer models
4mo ago→
N

Senior Quantum AI Research Scientist, Applied Research

NVIDIA·US, WA, Redmond; US, CA, Santa Clara·On-site
Senior·Research Scientist
PythonMachine learningQuantumcomputingAI
4mo ago→
A

Senior Machine Learning Engineer

Adobe·San Jose·On-site
Senior·ML Engineer·$152k – $265k
Machine learningAIPythonOrchestration
4mo ago→
JA

Senior Staff AI Engineer

JazzX AI·San Francisco Bay Area·On-site
Staff·AI Engineer·$207k – $290k
Large-scale ml systemsReinforcement learningDistributed computingModel optimization
4mo ago→
II

AI Researcher

Ivo Inc·San Francisco·On-site
·$180k – $200k
AILLMsAgentic AIRAG
4mo ago→
C

Machine Learning Intern/Co-op (Fall, 2026)

Cohere·Canada, Europe, United States, United Kingdom·Hybrid
Junior·ML Engineer
Machine learningPythonAIDeep learning
5mo ago→
RA

Senior Inference Optimization ML Engineer

Rhoda AI·Mountain View·On-site
Senior·ML Engineer·$175k – $250k
Inference optimizationMultimodal model deploymentHardware aware mlCloud edge performance tuning
5mo ago→
VF

Machine Learning Researcher

Virtu Financial·New York·On-site
Mid·Research Scientist·$200k – $300k
Deep learningFeature engineeringQuantitative modelingMarket data analysis
5mo ago→
U

Member of Technical Staff - Quantitative Research

Udio·New York City (Remote possible for exceptional candidates)·Hybrid
Mid·Research Scientist·$250k – $350k
Machine learningStatistical modelingData pipelinesMusic data analysis
5mo ago→
A

Machine Learning Engineer

Abakaai·Mountain View, CA·On-site
Mid·ML Engineer·$175k – $275k
PythonTensorFlowPyTorchAWS
5mo ago→
NS

Senior Computer Vision Engineer (Localization)

Niantic Spatial·San Francisco, CA, Sunnyvale, CA·On-site
Senior·AI Engineer·$189k – $255k
Computer vision localizationVisual positioning systemsGeospatial mappingReal world ai models
5mo ago→
BR

Machine Learning Lead (LLM)

Blue Rose Research·Remote·Remote
Senior·AI Engineer·$165k – $210k
LLMsGenerative AIPolitical campaign analyticsElection forecasting
5mo ago→
B

Senior Machine Learning Engineer

BJAK·United Kingdom·On-site
Staff·ML Engineer
LLMsWorkflow automationContext managementReliability engineering
5mo ago→
B

Principal Machine Learning Engineer

BJAK·United Kingdom·Hybrid
Staff·ML Engineer
Production ml systemsLong-running workflowsPersistent context handlingReliable ai deployment
5mo ago→
PL

ML Systems Engineer

Periodic Labs·Menlo Park, CA·On-site
·$250k – $350k
AIMachine learningJAXAgentic AI
5mo ago→
A

Machine Learning Research Engineer (MLRE) - Research

Achira·San Francisco Office, New York Office·Hybrid
·$165k – $259k
AIMachine learningSoftware engineeringAgentic AI
5mo ago→
A

Machine Learning Research Engineer (MLRE) - Workflows/Systems

Achira·San Francisco Office, New York Office·On-site
Senior·ML Engineer·$165k – $259k
PythonPyTorchDistributed computingGpu programming
5mo ago→
A

Machine Learning Research Engineer (MLRE) - GPUs

Achira·San Francisco Office, Remote, New York Office·Hybrid
·$165k – $259k
AIDeep learningMachine learningPyTorch
5mo ago→
S

Senior Staff Machine Learning Engineer, Content Platform

Spotify·London·On-site
Staff·ML Engineer
Distributed systemsLarge scale ml pipelinesContent classificationNlp for audio text
5mo ago→
L

AI Research Engineer, Post-Training

Lovable·Stockholm, London·On-site
AILLMsPyTorchJAX
5mo ago→
U

Senior Applied Scientist

UiPath·Bucharest·On-site
Senior·ML Engineer
Machine learningModel deploymentScalable mlProduction mlops
5mo ago→
N

Machine Learning Engineer - Humanoid Robotics

NVIDIA·China, Shanghai·On-site
Senior·ML Engineer
PythonMachine learningRoboticsComputer vision
5mo ago→
FH

Research Engineer

Forus Health·New York office·On-site
·$170k – $300k
AILLMsFine-tuningNatural language processing
5mo ago→
MR

Robotic AI Engineer/Applied Scientist - Foundation Models

Maven Robotics·San Francisco Bay Area, California USA·On-site
Senior·AI Engineer
Foundation modelsNeural architecture designReinforcement learningRobotics perception systems
5mo ago→
SA

Research Engineer, Post-training & Deployment

Skild AI·San Mateo, CA·On-site
·$100k – $300k
AIMachine learningDeep learningPython
5mo ago→

JAX jobs by city

San Francisco (89)New York (48)Mountain View (40)London (39)Seattle (22)Paris (16)

Skills often asked for alongside JAX

PyTorch (439)AI (407)Python (395)Machine Learning (388)Deep Learning (243)LLMs (217)TensorFlow (200)Software Engineering (194)

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.