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.

Research Engineer

Skild AI·Pittsburgh, San Mateo ·On-site

·$100k – $300k

AI Machine learning Python Deep learning

5mo ago→

Research Scientist

Skild AI·Pittsburgh, San Mateo ·On-site

Senior·Research Scientist·$100k – $300k

Reinforcement learning Computer vision Robotics Large scale ml

5mo ago→

Machine Learning Engineer, Reinforcement Learning

Skild AI·Pittsburgh, San Mateo·On-site

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

Reinforcement learning Algorithm design Model optimization Experimental design

5mo ago→

Machine Learning Engineer

Skild AI·San Mateo, Pittsburgh ·On-site

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

Reinforcement learning Algorithm design Model optimization Experimental validation

5mo ago→

Deep Learning Manipulation Engineer

Skild AI·Pittsburgh, San Mateo·On-site

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

Deep learning Reinforcement learning Robotics Computer vision

5mo ago→

Principal Deep Learning Communication Architect

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

Staff·MLOps

Deep learning Distributed systems Gpu programming Mpi

5mo ago→

Research Engineer, Model Inference & Serving - Paris

H Company·Hybrid Paris·Hybrid

Agentic AI AI Model serving Software engineering

5mo ago→

Research Engineer / Scientist, Post-training & Reinforcement Learning - London

H Company·Hybrid London·Hybrid

Agentic AI AI LLMs Fine-tuning

5mo ago→

Research Engineer, Model Inference & Serving - London

H Company·Hybrid London·Hybrid

Agentic AI AI Model serving Software engineering

6mo ago→

AI & Data Analytics Research Fellow-Radiation Oncology

Mayo US·Rochester, MN, United States·On-site

AI Data analytics Machine learning Statistics

6mo ago→

Internship - Search Machine Learning Engineer

Perplexity·London·On-site

Junior·ML Engineer

Retrieval augmentation Ranking algorithms Fine-tuning Search model evaluation

6mo ago→

Research Engineer, World Models

Waabi·Toronto, ON·On-site

·$155k – $269k

AI Generative AI LLMs Distributed training

6mo 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

6mo ago→

Research Scientist, Neural Reconstruction

Waabi·Toronto, ON·On-site

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

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

6mo ago→

Senior / Staff Software Engineer, ML Datasets & Data Pipelines

Waabi·Toronto, ON·On-site

·$148k – $260k

AI Data pipelines Deep learning Machine learning

6mo 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

6mo ago→

HPE Labs - Postdoctoral Research Scientist: Generative AI for Photonic and Analog Electronic Design Automation

Hewlett Packard Enterprise·Diegem, Flemish Brabant, Belgium·Hybrid

Senior·Research Scientist

Generative AI Photonic design automation Analog electronic cad Neural network architectures

6mo ago→

Research Engineer

Adobe·Seattle·On-site

·$121k – $239k

Machine learning Generative AI AI Computer vision

6mo ago→

Research Engineer

Nace AI·Palo Alto, CA·On-site

AI Agentic AI Machine learning LLMs

6mo ago→

Scientist, Machine Learning

Atomic AI·South San Francisco, California, United States·On-site

Senior·Research Scientist·$170k

Python PyTorch TensorFlow Deep learning

7mo ago→

Senior Machine Learning Engineer (LLMs)

Albi·Chicago, Illinois, United States·On-site

Senior·ML Engineer

Python PyTorch Transformers Lora

7mo ago→

AI Engineer

Airwallex·US - San Francisco·On-site

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

Machine learning Deep learning Python Tensorflow pytorch

7mo ago→

Data Engineer - Foundational

Harmattan AI·Paris·On-site

Mid·Data Engineer

Data pipelines Distributed systems Video data processing Etl workflows

7mo ago→

Principal Machine Learning Engineer

BJAK·China·Hybrid

AI Machine learning Fine-tuning Data pipelines

8mo ago→

AI Researcher – Multilingual Data

Featherlessai·Remote (world)·On-site

AI LLMs Natural language processing Data quality

8mo ago→

AI Researcher — AI Architecture Research

Featherlessai·Remote (world)·On-site

AI Machine learning Neural networks PyTorch

8mo ago→

Machine Learning Engineer — Distillation

Featherlessai·Remote (world)·Remote

Mid·ML Engineer

Knowledge distillation Teacher student models Model compression Large scale experimentation

8mo ago→

Machine Learning Engineer — AI Architecture Research

Featherlessai·Remote (world)·Remote

Senior·AI Engineer

Neural architecture design Transformer alternatives Scaling laws experiments Hybrid model development

8mo ago→

Machine Learning Engineer (Foundation Models & Personalization)

Eight Sleep·San Francisco·On-site

Mid·ML Engineer

Foundation models Personalization algorithms Deep learning Time series

8mo ago→

Internship - Search Machine Learning Engineer

Perplexity·Belgrade·On-site

Junior·ML Engineer

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

9mo ago→

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ·United States·Hybrid

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

LLMs Quantitative-modeling Generative AI Distributed-computing

9mo ago→

Machine Learning Scientist - ML Research

Arena Intelligence, Inc·Bay Area·On-site

Senior·Research Scientist

Machine learning Deep learning Model evaluation Experimental design

9mo ago→

(Data & AI/ML) Solutions Architect | Latam

Cuesta Partners·Bogotá, Barranquilla , Medellín, Cali·Hybrid

AI MLOps LLMs Data engineering

10mo ago→

Liquid Labs - Research Engineer

Liquid AI·Boston, Boston·Hybrid

AI Neural networks Python PyTorch

10mo ago→

Senior ML Systems Engineer, Frameworks & Tooling

Cohere·London, San Francisco, New York, Paris, Toronto, Montreal·Hybrid

·$295k – $535k

AI Distributed systems Machine learning LLMs

10mo ago→

Research Engineer / Scientist, Post-training - Paris

H Company·Hybrid Paris, Hybrid London·Hybrid

Agentic AI AI LLMs Reinforcement learning

11mo ago→

Machine Learning Engineer

Relace·San Francisco·On-site

Mid·ML Engineer

LLMs Model inference optimization Distributed systems Tokenization pipelines

11mo ago→

Machine Learning Scientist

Relace·San Francisco·On-site

Mid·ML Engineer

Transformer architectures Llm optimization Tokenization engineering RAG

11mo ago→

Research Scientist

Simular·Palo Alto, Singapore·On-site

Senior·Research Scientist

Reinforcement learning Multimodal reasoning Planning algorithms Ai safety

12mo ago→

AI / ML Engineer - Known

Pear VC·San Francisco·On-site

Mid·ML Engineer

LLMs Recommendation systems Production ml pipelines Conversational ai

12mo ago→

AI Research Scientist

Collate·San Francisco·On-site

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

Natural language processing Transformer models Biomedical nlp Deep learning

13mo ago→

Research Scientist

Dyna Robotics·Redwood City, CA·On-site

Senior·Research Scientist·$200k – $400k

Deep learning Computer vision Reinforcement learning Multimodal modeling

13mo ago→

Member of Technical Staff, Integration/RL Team (Research Engineer)

Cohere·Paris, London, New York, Toronto, Montreal·Hybrid

·$295k – $535k

AI Machine learning LLMs Data infrastructure

13mo 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

15mo ago→

Machine Learning Engineer - Enterprise

Boson AI·Toronto·On-site

Mid·ML Engineer

LLMs Agentic systems Enterprise ml deployment Fine-tuning

15mo ago→

Research Scientist

Lead Allies Inc·San Francisco·Hybrid

Senior·Research Scientist

Preference modeling Experimental design Ai model evaluation Human-in-the-loop ml

16mo ago→

AI Scientist - Machine Learning (KR/US)

Gausslabs·Palo Alto, CA / Yeoksam, Seoul·On-site

Senior·Research Scientist

Deep learning Temporal modeling Manufacturing Ml research

17mo ago→

ML Runtime Optimization Engineer

Applied Intuition·Sunnyvale·On-site

·$217k – $318k

AI Machine learning PyTorch JAX

19mo ago→

Machine Learning Engineers

for Humanity·San Jose, CALIFORNIA, United States·On-site

Senior·ML Engineer·$194k – $355k

Machine learning model development Privacy preserving techniques Large scale distributed systems Deep learning frameworks

24mo ago→

Quantitative Researcher

QuantCo·USA·On-site

Mid·Research Scientist

Machine learning Statistical modeling Causal inference Randomized experiments

29mo ago→

page 9 of 10 · 451 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 daily as roles open and close.

Who’s hiring for JAX

JAX jobs by city

Skills often asked for alongside JAX

Other skills in demand