updated august 2026

JAX Jobs in Data Science (August 2026)

150 open data science, ML and AI roles ask for JAX as of August 2026, across 66 hiring companies — 37 of them remote-friendly. Graphcore is hiring the most, with 10 openings. London is the busiest market. Every listing is scraped daily from the employer's own careers page or ATS, and links straight to the official application.

Senior Machine Learning Engineer (Large Systems)

Graphcore·London, UK·On-site

Senior·ML Engineer

AI Machine learning Software engineering Distributed training

4w ago→

Staff Engineer (ML Engineer)

Graphcore·Bristol, UK·On-site

Staff·ML Engineer

Machine learning frameworks Benchmarking Automated testing Quantization

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Cyprus·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow scikit-learn

4w ago→

Applied Scientist/Research Engineer, EMEA

Mistral AI·Paris, Amsterdam, London, Lausanne·Hybrid

AI PyTorch JAX Python

4w ago→

Staff Engineer (ML Engineer)

Graphcore·Cambridge, UK·On-site

Staff·ML Engineer

Machine learning frameworks Benchmarking Automated testing Quantization

4w ago→

Senior Machine Learning Engineer (Large Systems)

Graphcore·Bristol, UK·On-site

Senior·ML Engineer

AI Machine learning Software engineering Distributed training

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Serbia·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow Reinforcement learning

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Poland·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow pandas

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Georgia·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow Reinforcement learning

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Portugal·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow Reinforcement learning

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Spain·Hybrid

Senior·ML Engineer

Python PyTorch TensorFlow Reinforcement learning

4w ago→

Senior Engineer (ML Engineer)

Graphcore·Cambridge, UK·On-site

Senior·ML Engineer

Machine learning frameworks Benchmarking Quantization Distributed computing

4w ago→

Senior Engineer (ML Engineer)

Graphcore·London, UK·On-site

Senior·ML Engineer

Machine learning frameworks Benchmarking Quantization Distributed computing

4w ago→

Senior Engineer (ML Engineer)

Graphcore·Bristol, UK·On-site

Senior·ML Engineer

Machine learning frameworks Benchmarking Quantization Distributed computing

4w ago→

Senior Machine Learning Engineer (Large Systems)

Graphcore·Cambridge, UK·On-site

Senior·ML Engineer

Pytorch|tensorflow|jax Distributed training Hardware optimization Graph neural networks

4w ago→

Senior Machine Learning Engineer (Large Systems)

Graphcore·Gdańsk, Pomeranian Voivodeship, Poland·On-site

Staff·ML Engineer·$260k – $352k

Tensorflow/pytorch Distributed computing Ai infrastructure Graph neural networks

4w ago→

Staff Engineer (ML Engineer)

Graphcore·London, UK·On-site

Staff·ML Engineer

Machine learning frameworks Benchmarking Automated testing Quantization

4w ago→

Senior ML Engineer / ML Lead

MIRA CONSTRUCTION L.L.C·TELECOMMUTE, Cyprus; TELECOMMUTE, Georgia; TELECOMMUTE, Poland; TELECOMMUTE, Portugal; +2 more·Hybrid

Senior·ML Engineer

Reinforcement learning Quantitative trading High frequency trading Distributed computing

4w ago→

2027 Internship - Quantitative Researcher (Master or PhD)

Virtu Financial·Dublin, Ireland·On-site

Junior·Research Scientist

Python Statistical arbitrage High frequency trading Time series

4w ago→

Senior Research Scientist FMTA

DeepL·London·On-site

Senior·Research Scientist

Python Machine learning Natural language processing AI

4w ago→

Machine Learning Researcher

Rain·El Segundo, CA·On-site

Mid·Research Scientist

Machine learning model development Numerical weather prediction Atmospheric data processing Experimental design evaluation

1mo ago→

Sr. Software Engineer, Computer Vision

SpaceX·Hawthorne, CA·On-site

Senior·AI Engineer·$160k – $225k

Computer vision Deep learning Convolutional neural networks Model deployment

1mo ago→

Senior / Staff Machine Learning Engineer, Applied AI

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

Staff·ML Engineer·$180k – $298k

LLMs Model training evaluation Production ml systems Model adaptation

1mo ago→

AI Engineer

Arondite·London, England, United Kingdom·On-site

Mid·AI Engineer

Python Machine learning Distributed systems Real-time ai

1mo ago→

AI Engineer

Arondite·London, England, United Kingdom·On-site

Mid·AI Engineer

Python PyTorch TensorFlow Machine learning

1mo ago→

Senior Research Scientist | Multimodal Systems

DeepL·London·On-site

Senior·Research Scientist

Python Machine learning Natural language processing Deep learning

1mo ago→

Senior Research Scientist | Model Steering

DeepL·London·On-site

Senior·Research Scientist

Python Machine learning Natural language processing Deep learning

1mo ago→

Senior Research Scientist | Model Scaling

DeepL·London·On-site

Senior·Research Scientist

Python Machine learning Deep learning Natural language processing

1mo ago→

Deployed AI Engineer

Arondite·London, England, United Kingdom·On-site

Senior·AI Engineer

Python PyTorch TensorFlow Kubernetes

1mo ago→

AI Applied Scientist

Figma·San Francisco, CA • New York, NY • United States·On-site

·$153k – $376k

Machine learning AI Generative AI Fine-tuning

1mo ago→

Research Scientist (Machine Learning), London

Isomorphic Labs·London·On-site

Mid·Research Scientist

Deep learning Protein folding Generative AI Biomedical ml

1mo ago→

Staff Machine Learning Engineer

Xero·CAN: Vancouver (333 Seymour St), CAN: Vancouver (333 Seymour St)·Hybrid

Staff·ML Engineer·$195k – $235k

Generative AI Backend-ml Model-deployment Scalable-ml-systems

1mo ago→

Research Scientist (Applied LLMs), London

Isomorphic Labs·London·On-site

Mid·Research Scientist

LLMs Transformer architectures Protein folding Drug discovery

1mo ago→

R&D Machine Learning Engineer

DeepLab·Athens, Attica, Greece·On-site

Mid·ML Engineer

Pytorch tensorflow Deep learning research Computer vision Adtech systems

1mo ago→

Senior Deep Learning Systems Architect

NVIDIA·US, CA, Santa Clara·On-site

Senior·AI Engineer

Deep learning Hardware architecture Machine learning Data analytics

1mo ago→

R&D Machine Learning Engineer

DeepLab·Athens, Attica, Greece·On-site

Senior·ML Engineer

Python TensorFlow PyTorch Deep learning

1mo ago→

Applied Scientist

Mistral AI·Seoul·On-site

AI RAG PyTorch JAX

1mo ago→

Applied Scientist / Research Engineer, AI4Engineering

Mistral AI·Paris, Luxembourg, Amsterdam, Linz, Munich, London, Lausanne·Hybrid

AI LLMs Model evaluation RAG

1mo ago→

Research Engineer, Machine Learning

Mistral AI·Palo Alto, San Francisco·Hybrid

Mid·ML Engineer

LLMs Distributed training Reinforcement learning Model evaluation metrics

1mo ago→

Lead Applied Scientist

Mistral AI·Seoul·On-site

Senior·Research Scientist

LLMs Multimodal ai Computer vision Natural language processing

1mo ago→

AI Engineer

Titan OS·Barcelona, Catalonia, Spain·On-site

Mid·ML Engineer

Python TensorFlow PyTorch Recommendation systems

1mo ago→

AI Engineer, AI & Applications

Firmus Technologies·Singapore or Australia·On-site

Mid·AI Engineer

PyTorch Torch-titan Megatron-lm Fsdp

1mo ago→

Applied Scientist / Research Engineer (Internship)

Mistral AI·Paris, London·On-site

Junior·Research Scientist

LLMs Reinforcement learning Distributed computing Model optimization

1mo ago→

Research Engineer, Machine Learning

Mistral AI·Paris, Zurich, Warsaw, London·On-site

AI Data pipelines Deep learning Distributed training

1mo ago→

Computer Vision AI & ML Engineer

Skild AI·Bengaluru, India·On-site

Mid·ML Engineer

Computer vision Deep learning Convolutional neural networks Robotics perception

1mo ago→

Physics Informed Machine Learning Scientist

ASML·San Diego, CA, USA·On-site

Mid·Research Scientist·$135k – $203k

Physics-informed-neural-networks Laser-plasma-interactions Optical-systems-modeling Control-algorithm-development

1mo ago→

Staff Machine Learning Engineer

Adobe·San Jose·On-site

Staff·ML Engineer·$173k – $307k

Machine learning Natural language processing Deep learning Computer vision

2mo ago→

Senior Machine Learning Engineer

Field AI·Irvine, CA·On-site

Senior·ML Engineer

Transformer architectures Convolutional neural networks Reinforcement learning Robotics perception

2mo ago→

Senior Research Scientist, Battery Materials Simulation

SandboxAQ·United States·Hybrid

Senior·Research Scientist·$168k – $252k

Quantum-chemistry-simulation Dft-bse-modelling Machine-learning-for-material-science High-performance-computing

2mo ago→

Principal Machine Learning Engineer - Forecasting

Amgen·India - Hyderabad·On-site

Staff·ML Engineer

Time series forecasting Uncertainty estimation LLMs Agentic AI

2mo ago→

page 2 of 3 · 150 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 with the daily scrape.

Who’s hiring for JAX

JAX jobs by city

Skills often asked for alongside JAX

Other skills in demand