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.

Vice President, Artificial Intelligence & Machine Learning Engineer

NEW

Blackrock·Edinburgh, Scotland·On-site

Staff·AI Engineer

Python PyTorch TensorFlow Natural language processing

2d ago→

Machine Learning Research Intern - Summer 2027 - Sydney

NEW

IMC Financial Markets·Sydney, Australia·Hybrid

Junior·Data Scientist

Machine learning Quantitative analysis Statistics Data preprocessing

2d ago→

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 learning Deep learning Computer vision Sensor fusion

5d ago→

Director, Artificial Intelligence & Machine Learning Engineer

Blackrock·Edinburgh, Scotland·On-site

Senior·AI Engineer

Generative AI Machine learning Optimization-algorithms Statistical modeling

6d ago→

Applied AI Engineer

GSK·USA - Pennsylvania - Upper Providence; Cambridge 300 Technology Square·On-site

Mid·AI Engineer·$136k – $227k

Python Machine learning AI Responsible ai

6d ago→

Machine Learning Engineer - Recommendation Systems

xAI·Palo Alto, California·On-site

Senior·ML Engineer·$180k – $440k

Python PyTorch TensorFlow Collaborative filtering

6d ago→

Engineering Manager - Machine Learning (AI Products)

Xero·NZ: Auckland: Xero 4 (96 St Georges Bay Rd, Level 2 & 3), AU: Melbourne: (260 Burwood Rd), AU: Sydney (45 Clarence St)·On-site

Senior·MLOps

Machine learning MLOps Observability Platform engineering

6d ago→

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

Blackrock·New York, NY·On-site

Staff·AI Engineer

Generative AI Machine learning Optimization-algorithms Statistical modeling

6d ago→

Machine Learning Engineer

Twilio·Remote - Ireland·Remote

ML Engineer

Python TensorFlow PyTorch scikit-learn

6d ago→

Machine Learning Engineer

Twilio·Remote - Spain·Remote

ML Engineer

Python Machine learning Deep learning TensorFlow

6d ago→

Data Scientist

Fuse·San Leandro, California·On-site

Senior·Research Scientist·$145k – $190k

Python C++High-performance computing Magnetohydrodynamics

1w ago→

Machine Learning Engineer – Motion Planning & Prediction

Avride·Austin, Texas·On-site

Senior·ML Engineer

Python C++TensorFlow PyTorch

1w ago→

Senior Staff LLM Inference Engineer

d-Matrix·Santa Clara·Hybrid

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

Machine learning Inference Heterogeneous hardware Pathfinding

1w ago→

Principal, Machine Learning Engineer

Lila Sciences·San Francisco, CA USA·On-site

Staff·MLOps·$252k – $374k

Python PyTorch Docker Kubernetes

1w ago→

Product ML Engineer

Photoroom·Paris·Hybrid

Mid·ML Engineer

Python PyTorch TensorFlow scikit-learn

1w ago→

Senior Deep Learning Algorithm Engineer

NVIDIA·US, CA, Santa Clara; US, CA, Remote·Hybrid

Senior·ML Engineer

Deep learning Distributed training Model parallelism LLMs

1w ago→

Machine Learning Research Intern - Summer 2027 - Chicago

IMC Financial Markets·Chicago, United States·On-site

Junior·Data Scientist

Machine learning Quantitative analysis Statistics Dataset analysis

1w ago→

Principal Applied Scientist- AI

UiPath·Bellevue·On-site

Staff·Research Scientist·$218k – $265k

Foundation models Multimodal Agentic workflows Post-training

1w ago→

Graduate Machine Learning Researcher - Chicago

IMC Financial Markets·Chicago, United States·On-site

Junior·Research Scientist

Machine learning Statistics Data analysis Model deployment

1w ago→

Helix AI Engineer, Training Performance

Figure AI·San Jose, CA·On-site

Senior·AI Engineer·$200k – $400k

Distributed training Gpu optimization Model training Cluster topology

1w ago→

ML Engineer, Surrogate Modeling (Vehicle Engineering)

SpaceX·Hawthorne, CA·On-site

Senior·ML Engineer·$125k – $145k

Python PyTorch TensorFlow NumPy

1w ago→

Senior Staff Research Scientist | Voice

DeepL·London·On-site

Staff·Research Scientist

Python Natural language processing Machine learning Voiceprocessing

1w ago→

Machine Learning Software Engineer, Research

PhysicsX·London, United Kingdom·On-site

Mid·ML Engineer

Python PyTorch C++Cuda

2w ago→

Senior Machine Learning Engineer, Sentry Tower

Anduril Industries·Irvine, California, United States; Remote·Hybrid

Senior·ML Engineer·$220k – $330k

Computer vision Sensor fusion Real time ai Autonomy systems

2w ago→

Staff ML Engineer, Fine Tuning - Slack

Salesforce·Washington - Seattle; Georgia - Atlanta; California - San Francisco·On-site

Staff·ML Engineer·$197k – $314k

Fine-tuning LLMs Machine learning Python

2w ago→

Machine Learning Engineer

IMC Financial Markets·Hong Kong, Hong Kong; Sydney, Australia·Hybrid

Mid·ML Engineer

Machine learning Distributed computing Large scale datasets Low latency

2w ago→

Machine Learning Engineer

IMC Financial Markets·Amsterdam, Netherlands; London, United Kingdom·Hybrid

Mid·ML Engineer

Machine learning Distributed computing Large scale datasets Gpu acceleration

2w ago→

Senior Machine Learning Researcher

Ai2C Technologies·Athens, Attica, Greece·On-site

Senior·ML Engineer

Deep learning Reinforcement learning Distributed computing Model optimization

2w ago→

Senior Machine Learning Engineer

Cloudflare·Hybrid·Hybrid

Senior·ML Engineer

Tensorflow/pytorch Distributed training Reinforcement learning Network security ml

2w ago→

Senior Machine Learning Researcher

Ai2C Technologies·Athens, Attica, Greece·On-site

Senior·ML Engineer

Python PyTorch TensorFlow scikit-learn

2w ago→

Senior Engineering Manager - Machine Learning

Super·Romania·On-site

Senior·ML Engineer

Machine learning model development Predictive analytics Real time data processing Mlops pipelines

3w ago→

Senior Software Engineer, CUDA Deep Learning Systems

NVIDIA·US, CA, Santa Clara; US, TX, Austin; US, TX, Remote·Hybrid

Senior·AI Engineer

Cuda programming Deep learning architectures Distributed computing Hardware optimization

3w ago→

Software Engineer, CUDA Deep Learning Systems

NVIDIA·US, CA, Santa Clara; US, TX, Austin; US, TX, Remote·On-site

Senior·ML Engineer

Cuda Deep learning Python C++

3w ago→

Senior Machine Learning Software Engineer, Research

PhysicsX·London, United Kingdom·On-site

Senior·ML Engineer

Deep learning Multi physics simulation Physics informed nn High fidelity computational models

3w ago→

Senior Engineering Manager - Machine Learning

Super·Spain·On-site

Senior·ML Engineer

Machine learning model development Predictive analytics Personalization algorithms Fraud detection

3w ago→

Senior Engineering Manager - Machine Learning

Super·Netherlands·On-site

Senior·MLOps

MLOps Predictive modeling Fraud detection Real time ml

3w ago→

Senior Engineering Manager - Machine Learning

Super·United Kingdom·On-site

Senior·ML Engineer

Machine learning model development Predictive analytics Real time data processing Mlops pipelines

3w ago→

Machine Learning Engineer (Match Group AI)

Match Group·Seoul, South Korea·On-site

Mid·ML Engineer

Reinforcement learning Deep learning Natural language processing Distributed training pipelines

3w ago→

Applied Scientist

Mistral AI·Palo Alto, New York·Hybrid

AI RAG PyTorch JAX

3w ago→

Quantitative Researcher - Machine Learning

Point72·New York, London, or Hong Kong·Hybrid

Mid·Research Scientist

Machine learning Quantitative research Trading signal development Proprietary data analysis

3w ago→

Applied Scientist - All Levels

PhysicsX·Singapore ·On-site

AI Machine learning Deep learning Statistics

3w ago→

Research Engineer, Machine Learning (RL Velocity)

Anthropic·Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY·Remote

Mid·ML Engineer·$500k – $850k

Reinforcement learning Distributed systems Cloud infrastructure Performance optimization

3w ago→

Senior Software Engineer, CUDA Deep Learning Systems

NVIDIA·US, CA, Santa Clara; US, TX, Austin; US, TX, Remote·On-site

Senior·AI Engineer

Cuda Gpu programming Deep learning Distributed systems

3w ago→

Research Engineer, Machine Learning (Reinforcement Learning)

Anthropic·San Francisco, CA | New York City, NY·On-site

Mid·Research Scientist·$500k – $850k

Reinforcement learning LLMs Autonomous agent development Code generation rl

3w ago→

Research Engineer, Machine Learning (RL Velocity)

Anthropic·London, UK·On-site

Mid·MLOps·$370k – $630k

Reinforcement learning infrastructure Distributed system optimization Rl training pipelines Performance engineering

3w ago→

Research Engineer, Machine Learning (Reinforcement Learning)

Anthropic·London, UK·On-site

Mid·Research Scientist·$260k – $630k

Reinforcement learning LLMs Autonomous agents Code generation

3w ago→

Lead Research Scientist - Machine Learning (Clearance Required)

Systems Tech Research·Woburn, MA·On-site

Senior·Research Scientist·$174k – $220k

Machine learning Estimation theory Signal processing Predictive analytics

3w ago→

Lead Research Scientist - Machine Learning

Systems Tech Research·Woburn, MA·On-site

Senior·Research Scientist·$174k – $220k

Machine learning Estimation theory Signal processing Predictive analytics

3w ago→

Senior Computer Vision Engineer

Sword Health·Porto·On-site

Senior·AI Engineer

Computer vision models Deep learning Convolutional neural networks Image segmentation

3w ago→

Senior Research Scientist - Machine Leaning

Systems Tech Research·Woburn, MA·On-site

Senior·Research Scientist·$139k – $165k

Machine learning Estimation theory Signal processing Statistical modeling

3w ago→

page 1 of 3 · 150 jobs

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