◆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
LS

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 learningMeta learningUncertainty estimationMultimodal modeling
1mo ago→
W

Principal AI Researcher

Workday·USA, CA, Pleasanton; USA, GA, Atlanta; Canada, BC, Vancouver·On-site
Staff·Research Scientist
Machine learning researchAi platform architectureNatural language processingData modeling
1mo ago→
C

Senior AI Engineer

Centific·Redmond, Washington·On-site
·$160k – $220k
AILLMsRAGGenerative AI
1mo ago→
C

Principal Research Scientist – Vision AI, Simulation & Physical AI

Centific·Remote Work( USA)·Remote
·$200k – $300k
AILLMsRAGGenerative AI
1mo ago→
PI

ML Infra Engineer, Modeling

Physical Intelligence·San Francisco·On-site
AIJAXSoftware engineeringML infrastructure
1mo ago→
AI

Senior Research Engineer, Olmo + Molmo

Allen Institute·Seattle, WA·On-site
·$174k – $261k
Deep learningPythonAIGenerative AI
1mo ago→
N

ML Infrastructure Engineer

Nebius·Amsterdam, Netherlands; Remote - Europe; Remote - United States·Hybrid
Senior·MLOps
Distributed systemsGpu orchestrationMl inference optimizationCloud infrastructure
1mo ago→
A

AI Engineer

Armada·Bellevue Office, Sunset Corporate Campus·On-site
Mid·AI Engineer·$155k – $193k
Distributed ai systemsEdge computing infrastructureNvidia gpu optimizationMloops pipeline design
1mo ago→
AI

Director, Machine Learning, Virtual Cell Initiative

Arc Institute·Palo Alto, CA·On-site
Senior·Research Scientist·$380k – $420k
Deep learningBiological data modelingMulti-scale simulationHigh-performance computing
1mo ago→
N

Senior Deep Learning Frameworks CUDA Software Engineer

NVIDIA·US, CA, Santa Clara; US, TX, Austin; US, TX, Remote·Hybrid
Senior·AI Engineer
CudaPythonPyTorchJAX
1mo ago→
Q

Applied Scientist, AI/ML

QuantCo·USA·On-site
Senior·Research Scientist
Machine learningCausal inferenceMultimodal modelingFoundation models
1mo ago→
K

[AI Research Div.] Postdoctoral Researcher - LLMs (계약직)

KRAFTON·Seoul·On-site
Senior·Research Scientist
LLMsTransformer architecturesReinforcement learningNatural language processing
1mo ago→
A

Research Engineer, Pretraining Scaling - London

Anthropic·London, UK·On-site
Senior·ML Engineer·$260k – $630k
Large-scale distributed trainingModel pretraining optimizationDeep learning frameworksScalable ml infrastructure
1mo ago→
1L

Machine Learning Engineer

10a Labs·Remote·Remote
Senior·ML Engineer·$130k – $200k
Machine learningReinforcement learningModel evaluationNatural language processing
1mo ago→
A

Research Engineer

AfterQuery·San Francisco·On-site
·$210k – $450k
AIMachine learningML infrastructurePython
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·United Kingdom·Hybrid
Senior·MLOps
Distributed systems architectureLlm deploymentReliable ml pipelinesContext persistence
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·Seoul, Korea·On-site
Senior·ML Engineer
Production ml systemsLong-running workflowsPersistent context handlingModel deployment optimization
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·Singapore·On-site
Staff·ML Engineer
Distributed systemsLlm deploymentReliable ml pipelinesContext persistence
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·China·On-site
Senior·ML Engineer
Distributed systemsLlm integrationReliable ml pipelinesContext persistence
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·Indonesia·Hybrid
Senior·ML Engineer
Distributed systems architectureModel deployment pipelinesLong-running workflow reliabilityContext persistence
1mo ago→
B

Staff Machine Learning Engineer

BJAK·Indonesia·Hybrid
Staff·ML Engineer
Production ml systemsLong-running workflowsPersistent context handlingScalable ml pipelines
1mo ago→
B

Technical Lead, Machine Learning

BJAK·Indonesia·Hybrid
Senior·MLOps
Distributed systems architectureModel deployment pipelinesLong-context llm optimizationReliable workflow orchestration
1mo ago→
B

Lead Engineer, Machine Learning

BJAK·United States·Hybrid
Staff·ML Engineer
Distributed systems architectureModel deployment pipelinesLong-context llm optimizationReliable workflow orchestration
1mo ago→
B

Principal Machine Learning Engineer

BJAK·Indonesia·Hybrid
Staff·ML Engineer
Large language model integrationProduction ml pipelinesReliable llm deploymentContext management systems
1mo ago→
G

Director, Lead Solution Architect AI/ML

Gdit·USA VA Herndon·On-site
·$225k – $305k
Data sciencePyTorchAIMachine learning
1mo ago→
NR

Senior/Staff Software Engineer, Infrastructure (ML)

Nimble Robotics·San Francisco, CA·On-site
Staff·AI Engineer·$210k – $300k
Distributed systemsRobotics infrastructureAutonomous systemsSupply chain automation
1mo ago→
A

Senior AI Engineer (Risk & Payments)

Airwallex·AU - Sydney·On-site
AIAgentic AILangChainLlamaIndex
1mo ago→
A

Senior AI Engineer (Risk & Payments)

Airwallex·SG - Singapore·On-site
AIAgentic AILangChainLlamaIndex
1mo ago→
C

Senior AI/ML Specialist

Cyncly·Kochi, Kerala, India·On-site
LLMsFine-tuningRLHFHugging Face
1mo ago→
I

Senior Machine Learning Engineer II, Ads Response Prediction

Instacart·United States - Remote·Hybrid
Senior·ML Engineer·$240k – $254k
Machine learningResponse predictionFeature engineeringModel deployment
1mo ago→
B

Vice President, Artificial Intelligence & Machine Learning Engineer

Blackrock·Edinburgh, Scotland·On-site
Staff·AI Engineer
PythonPyTorchTensorFlowNatural language processing
1mo ago→
IF

Machine Learning Research Intern - Summer 2027 - Sydney

IMC Financial Markets·Sydney, Australia·Hybrid
Junior·Data Scientist
Machine learningQuantitative analysisStatisticsData preprocessing
1mo ago→
TL

Research Engineer

Tessera Labs·San Jose Office (HQ), New York City Office·On-site
·$200k – $300k
AISnowflakeReinforcement learningAgentic AI
1mo ago→
FP

Principal Scientist, Machine Learning - Biomolecules

Flagship Pioneering, Inc·Cambridge, MA USA·On-site
Staff·Research Scientist·$208k – $286k
Deep learning for biomolecular dataProtein structure predictionHigh-performance computational modelingAi-driven drug discovery pipelines
1mo ago→
FP

Principal Scientist, Machine Learning

Flagship Pioneering, Inc·Cambridge, MA USA·On-site
Staff·Research Scientist·$208k – $286k
Deep learningPharmaceutical r&d pipelinesBiomedical data modelingComputational biology
1mo ago→
A

Research Engineer, Pretraining Scaling

Anthropic·San Francisco, CA·On-site
Mid·ML Engineer·$350k – $850k
Large-scale distributed trainingModel pretraining optimizationDeep learning frameworksScalable ml infrastructure
1mo ago→
A

Research Engineer, Interpretability

Anthropic·San Francisco, CA·On-site
Mid·Research Scientist·$315k – $560k
Neural network interpretabilityMechanistic analysisModel reverse-engineeringAttention mechanism dissection
1mo ago→
A

Research Engineer, Discovery

Anthropic·San Francisco, CA·On-site
Mid·AI Engineer·$350k – $850k
Large language model infrastructureDistributed systems optimizationModel training pipelinesScientific reasoning benchmarks
1mo ago→
AI

Staff Gen AI Research Scientist 

Anduril Industries·Costa Mesa, California, United States; Seattle, Washington, United States; Washington, District of Columbia, United States·On-site
Staff·Research Scientist·$220k – $292k
Machine learningDeep learningComputer visionSensor fusion
1mo ago→
M

Senior Data Scientist

Meredith·New York- NY 225 Liberty; Remote US·Hybrid
Senior·Data Scientist·$175k – $190k
Recommendation systemsCollaborative filteringMatrix factorizationDeep learning
1mo ago→
K

Algorithm Engineer - Deep Learning

Kla·Milpitas, CA·On-site
Mid·ML Engineer·$136k – $232k
Deep learning frameworksAlgorithm optimizationSemiconductor process modelingComputer vision
1mo ago→
A

Staff Research Scientist, AI-Hardware Co-Design

Analogdevices·US, MA, Boston·On-site
·$172k – $237k
AIMachine learningDeep learningPyTorch
1mo ago→
H

AI Research Engineer - ML Engineering

Helsing·Berlin; London; Munich·On-site
Mid·ML Engineer
Machine learning frameworksDefense ai applicationsModel deployment pipelinesEthical ai considerations
1mo ago→
H

AI Research Engineer - Foundation Models

Helsing·Berlin; London; Munich; Paris·On-site
Senior·Research Scientist
Foundation modelsLLMsMachine learning researchDefense ai applications
1mo ago→
LA

Senior Data Scientist (US)

Lynx Analytics·New York, USA; Philadelphia, Pennsylvania, United States; San Francisco, California, United States·On-site
Senior·Data Scientist
PythonSQLStatistical modelingGenerative AI
1mo ago→
B

Artificial Intelligence & Machine Learning Engineer, Vice President - AI Labs

Blackrock·New York, NY·On-site
Staff·AI Engineer
Generative AIMachine learningOptimization-algorithmsStatistical modeling
1mo ago→
B

Director, Artificial Intelligence & Machine Learning Engineer

Blackrock·Edinburgh, Scotland·On-site
Senior·AI Engineer
Generative AIMachine learningOptimization-algorithmsStatistical modeling
1mo ago→
T

Machine Learning Engineer

Twilio·Remote - Ireland·Remote
ML Engineer
PythonTensorFlowPyTorchscikit-learn
1mo ago→
T

Machine Learning Engineer

Twilio·Remote - Spain·Remote
ML Engineer
PythonMachine learningDeep learningTensorFlow
1mo ago→
X

Machine Learning Engineer - Recommendation Systems

xAI·Palo Alto, California·On-site
Senior·ML Engineer·$180k – $440k
PythonPyTorchTensorFlowCollaborative filtering
1mo 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.