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.

Machine Learning Engineer

Zillow·Remote-Germany·Remote

Senior·ML Engineer

Python TensorFlow Computer vision Deep learning

2mo ago→

Staff AI Engineer

Airwallex·US - San Francisco·On-site

Staff·AI Engineer·$182k – $310k

Python Machine learning Deep learning Distributed systems

2mo ago→

Sr Applied Data Scientist - Item Recommendations (applied ML, PyTorch, ML Ops)

Target·7000 Target Pkwy N,NCD-0375 Brooklyn Park,MN 55445·On-site

Senior·Data Scientist·$98k – $176k

Python PyTorch Machine learning MLOps

2mo ago→

Sr Applied Data Scientist - Personalization (applied ML, PyTorch, RecSys)

Target·7000 Target Pkwy N,NCD-0375 Brooklyn Park,MN 55445·On-site

Senior·Data Scientist·$98k – $176k

Python PyTorch Recommendation systems Machine learning

2mo ago→

AI research scientist

Writer·San Francisco, CA, Seattle, WA, New York City, NY·Hybrid

Senior·Research Scientist·$234k – $349k

Python Machine learning Deep learning Natural language processing

2mo ago→

AIML Engineer, AI for Science

GSK·UK - Kings Cross – London; Heidelberg; Switzerland - Zug·On-site

Mid·ML Engineer·$84k – $140k

Deep learning Reinforcement learning High performance computing Biomedical data analysis

2mo ago→

Senior AI/ML Research Engineer (Computer Vision)

Intuitive Surgical·Sunnyvale, CA, United States·On-site

Senior·Research Scientist

Computer vision models Deep learning Real-time perception Multimodal ai

2mo ago→

Scientist /Senior Scientist, Multimodal & Relational Machine Learning Foundation Models

Altos Labs·San Francisco Bay Area, CA;San Diego, CA·On-site

Senior·Research Scientist·$201k – $258k

Multimodal learning Relational ml Foundation models Biological data processing

2mo ago→

Research Scientist - Salesforce AI Research

Salesforce·California - Palo Alto; California - San Francisco·On-site

·$117k – $314k

Software engineering Agentic AI LLMs Reinforcement learning

2mo ago→

Senior Machine Learning Engineer, End‑to‑End Autonomous Driving

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

Senior·ML Engineer

Python TensorFlow PyTorch Computer vision

2mo ago→

Senior Deep Learning Performance Architect

NVIDIA·US, CA, Santa Clara; US, WA, Redmond·On-site

Senior·AI Engineer

Deep learning architecture Performance analysis Ppa optimization C++ simulation

2mo ago→

Computer Vision AI & ML Engineer

Skild AI·San Mateo, CA·On-site

AI Machine learning Computer vision Deep learning

2mo ago→

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ·Canada·Hybrid

Senior·ML Engineer·$140k – $210k

LLMs Quantitative-modeling Deep-learning-frameworks Distributed-computing

3mo ago→

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ·Canada·Hybrid

Staff·ML Engineer·$172k – $258k

LLMs Quantitative-modeling Deep learning Distributed systems

3mo ago→

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ·United States·Hybrid

Senior·ML Engineer·$164k – $246k

LLMs Quantitative-modeling Deep learning Reinforcement learning

3mo ago→

Senior Quantum AI Research Scientist, Applied Research

NVIDIA·US, WA, Redmond; US, CA, Santa Clara·On-site

Senior·Research Scientist

Python Machine learning Quantumcomputing AI

3mo ago→

Staff Data Scientist | NLP

Machinify *·Remote - US·Remote

Staff·Data Scientist·$180k – $230k

Natural language processing Text classification Information extraction Transformer models

3mo ago→

AI engineer

Writer·San Francisco, CA; New York City, NY·Hybrid

Senior·AI Engineer·$187k – $329k

Python AI Machine learning LLMs

3mo ago→

Senior Machine Learning Engineer

Adobe·San Jose·On-site

Senior·ML Engineer·$152k – $265k

Machine learning AI Python Orchestration

3mo ago→

AI engineer (UK)

Writer·London, UK·On-site

Senior·AI Engineer

Python AI Machine learning LLMs

3mo ago→

Quantitative Researcher (Fresh STEM PhD graduates are welcome)

Binance·Hong Kong·On-site

Junior·Research Scientist

Machine learning Quantitative trading Algorithm design Time series forecasting

3mo ago→

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

Cohere·Canada, Europe, United States, United Kingdom·Hybrid

Junior·ML Engineer

Machine learning Python AI Deep learning

3mo ago→

Machine Learning Researcher

Virtu Financial·New York·On-site

Mid·Research Scientist·$200k – $300k

Deep learning Feature engineering Quantitative modeling Market data analysis

3mo ago→

Member of Technical Staff - Quantitative Research

Udio·New York City (Remote possible for exceptional candidates)·Hybrid

Mid·Research Scientist·$250k – $350k

Machine learning Statistical modeling Data pipelines Music data analysis

3mo ago→

Senior Inference Optimization ML Engineer

Rhoda AI·Mountain View·On-site

Senior·ML Engineer·$175k – $250k

Inference optimization Multimodal model deployment Hardware aware ml Cloud edge performance tuning

3mo ago→

Machine Learning Engineer, Global Public Sector

Scale AI·Doha, Qatar; London, UK·On-site

Senior·ML Engineer

Python LLMs Deep learning Ai safety

3mo ago→

Senior Staff Machine Learning Engineer, Content Platform

Spotify·London·On-site

Staff·ML Engineer

Distributed systems Large scale ml pipelines Content classification Nlp for audio text

3mo ago→

Machine Learning Engineer - Humanoid Robotics

NVIDIA·China, Shanghai·On-site

Senior·ML Engineer

Python Machine learning Robotics Computer vision

4mo ago→

Senior Applied Scientist

UiPath·Bucharest·On-site

Senior·ML Engineer

Machine learning Model deployment Scalable ml Production mlops

4mo ago→

Research Scientist

Skild AI·Pittsburgh, San Mateo ·On-site

·$100k – $300k

AI Machine learning Python Deep learning

4mo ago→

Deep Learning Manipulation Engineer

Skild AI·Pittsburgh, San Mateo·On-site

Mid·ML Engineer·$100k – $300k

Deep learning Reinforcement learning Robotics Computer vision

4mo ago→

Senior AI Engineer

Yuno·Hyderabad·On-site

Senior·AI Engineer

LLMs Agentic ai systems Payment fraud detection Distributed training

4mo ago→

Machine Learning Engineer, Reinforcement Learning

Skild AI·Pittsburgh, San Mateo·On-site

·$100k – $300k

AI Machine learning Reinforcement learning Python

4mo ago→

Machine Learning Engineer

Skild AI·San Mateo, Pittsburgh ·On-site

·$100k – $300k

AI Machine learning Reinforcement learning Python

4mo ago→

Internship - Search Machine Learning Engineer

Perplexity·London·On-site

Junior·ML Engineer

Retrieval augmentation Ranking algorithms Fine-tuning Search model evaluation

4mo ago→

Research Scientist, Neural Reconstruction

Waabi·Toronto, ON·On-site

Mid·Research Scientist·$155k – $269k

Neural-rendering 3d-reconstruction Sensor-fusion Deep learning

5mo ago→

Research Scientist, World Models

Waabi·Toronto, ON·On-site

Mid·Research Scientist·$155k – $269k

World models 4d environment reasoning Temporal reasoning Large scale simulation

5mo ago→

Machine Learning Research Engineer

Etched.ai·Cupertino, CA·On-site

Mid·ML Engineer

Transformer architecture optimization Asics hardware acceleration Low latency ml inference Mathematical operation translation

5mo ago→

AI Engineer

Airwallex·US - San Francisco·On-site

Mid·AI Engineer·$120k – $200k

Machine learning Deep learning Python Tensorflow pytorch

5mo ago→

Data Engineer - Foundational

Harmattan AI·Paris·On-site

Mid·Data Engineer

Data pipelines Distributed systems Video data processing Etl workflows

5mo ago→

Machine Learning Engineer (Foundation Models & Personalization)

Eight Sleep·San Francisco·On-site

Mid·ML Engineer

Foundation models Personalization algorithms Deep learning Time series

7mo ago→

Internship - Search Machine Learning Engineer

Perplexity·Belgrade·On-site

Junior·ML Engineer

Retrieval-augmented-search Ranking-algorithms LLMs Model-evaluation-metrics

7mo ago→

Senior Deep Learning Performance Architect

NVIDIA·US, CA, Santa Clara; US, WA, Redmond·On-site

Senior·ML Engineer

Deep learning Performance analysis Architecture AI

7mo ago→

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ·United States·Hybrid

Staff·ML Engineer·$196k – $294k

LLMs Quantitative-modeling Generative AI Distributed-computing

7mo ago→

Machine Learning Engineer

Relace·San Francisco·On-site

Mid·ML Engineer

LLMs Model inference optimization Distributed systems Tokenization pipelines

10mo ago→

Machine Learning Scientist

Relace·San Francisco·On-site

Mid·ML Engineer

Transformer architectures Llm optimization Tokenization engineering RAG

10mo ago→

Applied Machine Learning Engineer

Cohere·London, Edinburgh, Paris, Berlin·Hybrid

AI Machine learning LLMs Python

11mo ago→

AI Research Scientist

Collate·San Francisco·On-site

Mid·Research Scientist·$391k – $215k

Natural language processing Transformer models Biomedical nlp Deep learning

11mo ago→

Senior Machine Learning Engineer (Match Group AI)

Match Group·Seoul, South Korea·On-site

Senior·ML Engineer

Deep learning Reinforcement learning Model deployment Mloops pipelines

14mo ago→

Machine Learning Engineer - Enterprise

Boson AI·Toronto·On-site

Mid·ML Engineer

LLMs Agentic systems Enterprise ml deployment Fine-tuning

14mo ago→

page 3 of 3 · 150 jobs

← Previous

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