◆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

Research Engineer, Machine Learning (Reinforcement Learning)

Anthropic·London, UK·On-site
Mid·Research Scientist·$260k – $630k
Reinforcement learningLLMsAutonomous agentsCode generation
2mo ago→
II

Senior AI Researcher

Ivo Inc·San Francisco·On-site
·$200k – $325k
AILLMsAgentic AIRAG
2mo ago→
ST

Senior Research Scientist - Machine Leaning

Systems Tech Research·Woburn, MA·On-site
Senior·Research Scientist·$139k – $165k
Machine learningEstimation theorySignal processingStatistical modeling
2mo ago→
ST

Lead Research Scientist - Machine Learning (Clearance Required)

Systems Tech Research·Woburn, MA·On-site
Senior·Research Scientist·$174k – $220k
Machine learningEstimation theorySignal processingPredictive analytics
2mo ago→
ST

Lead Research Scientist - Machine Learning

Systems Tech Research·Woburn, MA·On-site
Senior·Research Scientist·$174k – $220k
Machine learningEstimation theorySignal processingPredictive analytics
2mo ago→
O

Machine Learning Performance Engineer

Optiver·New York, New York·On-site
Mid·MLOps·$200k
Deep learning frameworksScalable pipelinesPerformance optimizationDistributed computing
2mo ago→
G

AI Research Engineer

Graphcore·Bristol, UK·On-site
Mid·AI Engineer
Neural network architectureAi compute optimizationSemiconductor ai integrationLarge scale distributed training
2mo ago→
G

AI Research Engineer

Graphcore·Cambridge, UK·On-site
Senior·AI Engineer
Deep learning frameworksAi hardware optimizationNeural architecture designDistributed computing
2mo ago→
G

AI Research Engineer

Graphcore·London, UK·On-site
Mid·AI Engineer
Neural network architectureAi compute optimizationSemiconductor ai integrationLarge scale distributed training
2mo ago→
MA

Applied Scientist/Research Engineer, EMEA

Mistral AI·Paris, Amsterdam, London, Lausanne·On-site
Senior·Research Scientist
LLMsDistributed systemsCustom ai architectureEnterprise ai integration
2mo ago→
G

Senior Machine Learning Engineer (Large Systems)

Graphcore·Gdańsk, Pomeranian Voivodeship, Poland·On-site
Staff·ML Engineer·$260k – $352k
Tensorflow/pytorchDistributed computingAi infrastructureGraph neural networks
2mo ago→
G

Senior Machine Learning Engineer (Large Systems)

Graphcore·Bristol, UK·On-site
Senior·ML Engineer
AIMachine learningSoftware engineeringDistributed training
2mo ago→
G

Senior Machine Learning Engineer (Large Systems)

Graphcore·Cambridge, UK·On-site
Senior·ML Engineer
Pytorch|tensorflow|jaxDistributed trainingHardware optimizationGraph neural networks
2mo ago→
G

Senior Machine Learning Engineer (Large Systems)

Graphcore·London, UK·On-site
Senior·ML Engineer
AIMachine learningSoftware engineeringDistributed training
2mo ago→
VF

2027 Internship - Quantitative Researcher (Master or PhD)

Virtu Financial·Dublin, Ireland·On-site
Junior·Research Scientist
PythonStatistical arbitrageHigh frequency tradingTime series
2mo ago→
S

Post-Doctoral AI Researcher (2 year Fixed-Term)

Snowflake·US-WA-Bellevue·Hybrid
·$160k – $220k
SnowflakeAgentic AIAIMachine learning
2mo ago→
AS

Applied AI Data Scientist (Brazil)

AE Studio·Florianopolis office/ Brazil remote·Hybrid
Data Scientist
Machine learningData analysisResearchAi alignment
2mo ago→
AS

Applied AI Data Scientist

AE Studio·LA office / US remote·Hybrid
Research Scientist·$180k – $240k
Machine learningPythonDeep learningData analysis
2mo ago→
LS

Senior / Staff Machine Learning Engineer, Applied AI

Lila Sciences·Cambridge, MA USA; San Francisco, CA USA·On-site
Staff·ML Engineer·$180k – $298k
LLMsModel training evaluationProduction ml systemsModel adaptation
2mo ago→
S

Sr. Software Engineer, Computer Vision

SpaceX·Hawthorne, CA·On-site
Senior·AI Engineer·$160k – $225k
Computer visionDeep learningConvolutional neural networksModel deployment
2mo ago→
S

Research Scientist

Samsung·50/60 Station Road, Cambridge, United Kingdom·On-site
Senior·Research Scientist
Machine learning algorithmsModel co designEdge aiAdaptive inference
2mo ago→
B

Applied AI Engineer

BJAK·China·On-site
Mid·AI Engineer
LLMsWorkflow automationContext persistenceReliability engineering
2mo ago→
F

AI Applied Scientist

Figma·San Francisco, CA • New York, NY • United States·Hybrid
Senior·Research Scientist·$153k – $376k
Machine learningAi model developmentComputer visionNatural language processing
2mo ago→
Z

Software Engineer, ML Platform (ML Training)

Zoox·Foster City, CA·On-site
Data scienceDeep learningReinforcement learningAI
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Ireland·On-site
Staff·ML Engineer
Large language model integrationWorkflow automationContext persistenceReliability engineering
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Germany·On-site
Staff·ML Engineer
Large language model integrationWorkflow automationContext persistenceReliability engineering
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Sweden·On-site
Staff·AI Engineer
Large language model integrationWorkflow automationContext persistenceReliability engineering
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Portugal·Hybrid
Staff·AI Engineer
Large language model integrationWorkflow automationContext persistenceReliability engineering
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Spain·On-site
Staff·ML Engineer
Large language model integrationWorkflow automationContext persistenceReliability engineering
2mo ago→
B

Senior Machine Learning Engineer

BJAK·Poland·On-site
Staff·AI Engineer
Large language model integrationWorkflow automationContext managementReliability engineering
2mo ago→
S

Senior ML Research Engineer, Virtual Cell

SandboxAQ·Canada·Hybrid
·$148k – $222k
AISecurityMachine learningData infrastructure
2mo ago→
S

Senior ML Research Engineer, Virtual Cell

SandboxAQ·United Kingdom·On-site
·$84k – $126k
AISecurityMachine learningData infrastructure
2mo ago→
IL

ML Research Engineer, London

Isomorphic Labs·London·On-site
Mid·Research Scientist
Machine learning researchDrug discovery mlFrontier ai modelsScientific data pipelines
2mo ago→
IL

Research Scientist (Machine Learning), London

Isomorphic Labs·London·On-site
Mid·Research Scientist
Deep learningProtein foldingGenerative AIBiomedical ml
2mo ago→
IL

Research Scientist (Applied LLMs), London

Isomorphic Labs·London·On-site
Mid·Research Scientist
LLMsTransformer architecturesProtein foldingDrug discovery
2mo ago→
S

Senior ML Research Engineer, Virtual Cell

SandboxAQ·United States·Hybrid
·$168k – $252k
AISecurityMachine learningData infrastructure
2mo ago→
D

R&D Machine Learning Engineer

DeepLab·Athens, Attica, Greece·On-site
Mid·ML Engineer
Pytorch tensorflowDeep learning researchComputer visionAdtech systems
2mo ago→
N

Senior Deep Learning Systems Architect

NVIDIA·US, CA, Santa Clara·On-site
Senior·AI Engineer
Deep learningHardware architectureMachine learningData analytics
2mo ago→
O

Research Engineer, Privacy

OpenAI·San Francisco·Hybrid
Machine learningDeep learningPyTorchJAX
2mo ago→
MA

AI Scientist

Mistral AI·Paris, London·On-site
AILLMsDistributed systemsPyTorch
3mo ago→
MA

AI Scientist

Mistral AI·Zurich·On-site
AILLMsPythonRust
3mo ago→
MA

Lead Applied Scientist

Mistral AI·Seoul·On-site
Senior·Research Scientist
LLMsMultimodal aiComputer visionNatural language processing
3mo ago→
MA

Applied Scientist

Mistral AI·Seoul·On-site
Senior·Research Scientist
LLMsMultimodal aiState-of-the-art model developmentEnterprise ai integration
3mo ago→
HA

Senior Software Engineer - Backend & Machine Learning

Hume AI·New York, New York, United States·On-site
Senior·AI Engineer·$180k – $250k
Backend service developmentML infrastructureAudio processing modelsModel evaluation tools
3mo ago→
MA

AI Scientist

Mistral AI·Palo Alto, New York, San Francisco·Hybrid
AILLMsMachine learningNatural language processing
3mo ago→
MA

Applied Scientist / Research Engineer (Internship)

Mistral AI·Paris, London·On-site
Junior·Research Scientist
LLMsReinforcement learningDistributed computingModel optimization
3mo ago→
MA

Research Engineer, Machine Learning

Mistral AI·Paris, Zurich, Warsaw, London·Hybrid
Mid·AI Engineer
Machine learning model developmentDistributed computingClean code engineeringAi infrastructure
3mo ago→
R

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

Roche·South San Francisco, California, United States of America·On-site
·$148k – $274k
AILLMsReinforcement learningMachine learning
3mo ago→
C

Matterport – Senior Machine Learning/Computer Vision Engineer – 3D Reconstruction and Semantic Understanding

Costar·Sunnyvale (US)·On-site
·$190k – $253k
Machine learningComputer visionDeep learningPython
3mo ago→
XT

AI Scientist - Biomedical Multimodal Modeling

Xaira Therapeutics·South San Francisco, California, United States·On-site
·$170k – $240k
AIGenerative AIDeep learningPython
3mo 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.