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.

LLM Application Engineer

BJAK·Sweden·Hybrid

Mid·AI Engineer

LLMs Workflow automation Context management Reliability engineering

2mo ago→

LLM Application Engineer

BJAK·Poland·Hybrid

Mid·AI Engineer

LLMs Proactive ai workflows Persistent context handling Task completion systems

2mo ago→

LLM Application Engineer

BJAK·Spain·Hybrid

Mid·AI Engineer

LLMs Proactive ai workflows Context management Task completion systems

2mo ago→

LLM Application Engineer

BJAK·Portugal·Hybrid

Mid·AI Engineer

LLMs Proactive ai workflows Persistent context handling Task completion systems

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

2mo ago→

Senior Machine Learning Engineer, Agentic Science/Generative Models, AI for Biology & Translation (AIBT)

Roche·South San Francisco, California, United States of America·On-site

·$168k – $312k

AI Agentic AI MLOps Machine learning

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

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

2mo ago→

Machine Learning Engineer, Perception

Wing·Palo Alto, California·On-site

·$190k – $202k

Machine learning Deep learning Computer vision Python

2mo ago→

BioMedical AI Research Engineer

Xaira Therapeutics·South San Francisco, California, United States·On-site

·$227k – $295k

AI Generative AI Agentic AI LLMs

2mo ago→

Senior Machine Learning Scientist I/II, Model-Driven Optimization

Generate Biomedicines·Somerville, MA·On-site

Senior·Research Scientist·$192k – $265k

Sequential decision-making algorithms Experimental design optimization Protein property modeling Closed-loop systems

2mo ago→

Senior Machine Learning Engineer

Cloudflare·Hybrid·Hybrid

Senior·ML Engineer

Tensorflow/pytorch Distributed training Reinforcement learning Network security ml

2mo ago→

Senior Research Engineer - Autonomous Vehicles

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

Senior·Research Scientist

Deep learning Reinforcement learning Generative AI End to end av architectures

2mo ago→

Reinforcement Learning AI Engineer

Booz Allen Hamilton Holding·Huntsville, AL; McLean, VA; El Segundo, CA; Colorado Springs, CO·On-site

·$99k – $225k

Reinforcement learning AI Data science Machine learning

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

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

2mo ago→

Senior Machine Learning Perception Engineer - Fallback Driving System

General Motors·GM Automation - Sunnyvale - GM Automation - Sunnyvale·Hybrid

·$171k – $261k

Machine learning Software engineering Deep learning PyTorch

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·China·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflows Context-persistence Task-automation

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Ireland·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflows Persistent-context-handling Task-completion-ml

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Germany·On-site

Staff·AI Engineer

Production-ml-systems Long-running-workflow-reliability Persistent-context-handling Task-completion-ml

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Sweden·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflows Context-persistence Task-automation

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Portugal·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflows Context-persistence Task-completion-systems

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Spain·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflow-reliability Context-persistence Hallucination-mitigation

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Poland·Hybrid

Staff·ML Engineer

Production-ml-systems Long-running-workflows Context-persistence Task-automation

2mo ago→

Applied AI Engineer

BJAK·Ireland·Hybrid

Mid·AI Engineer

Fine-tuning Workflow-automation Context-persistence Reliability-engineering

2mo ago→

Applied AI Engineer

BJAK·Germany·On-site

Mid·AI Engineer

Fine-tuning Workflow-automation Context-persistence Reliability-engineering

2mo ago→

Applied AI Engineer

BJAK·Portugal·Hybrid

Mid·AI Engineer

Fine-tuning Workflow-automation Context-persistence Reliability-engineering

2mo ago→

Applied AI Engineer

BJAK·Spain·Hybrid

Mid·AI Engineer

Fine-tuning Workflow-automation Context-persistence Prompt engineering

2mo ago→

Senior Machine Learning Engineer

BJAK·Zurich, Switzerland·On-site

Staff·ML Engineer

Large language model integration Workflow automation Context persistence Reliability engineering

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Zurich, Switzerland·On-site

Staff·ML Engineer

LLMs Workflow automation Context management Reliability engineering

2mo ago→

Applied AI Engineer

BJAK·Zurich, Switzerland·On-site

Mid·AI Engineer

LLMs Workflow automation Context management Reliability engineering

2mo ago→

Senior Machine Learning Engineer

BJAK·Seoul, Korea·On-site

Staff·ML Engineer

LLMs Workflow automation Context persistence Reliability engineering

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·United Kingdom·Hybrid

Staff·AI Engineer

Production ml pipelines Long context llms Task completion systems Hallucination mitigation

2mo ago→

Applied AI Engineer

BJAK·Singapore·On-site

Mid·AI Engineer

Llm-integration Workflow-automation Context-persistence Reliability-engineering

2mo ago→

Applied AI Engineer

BJAK·United States·Hybrid

Mid·AI Engineer

Llm prompt engineering Workflow automation Context persistence Reliability architecture

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Singapore·On-site

Staff·ML Engineer

Production-ml-systems Long-running-workflows Persistent-context-handling Task-completion-ml

2mo ago→

Senior Machine Learning Engineer

BJAK·Singapore·On-site

Staff·ML Engineer

Large language model integration Workflow automation Context persistence Reliability engineering

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·Seoul, Korea·On-site

Staff·ML Engineer

Python PyTorch Machine learning Natural language processing

2mo ago→

Applied AI Engineer

BJAK·Seoul, Korea·On-site

Mid·AI Engineer

Python PyTorch Fine-tuning Prompt engineering

2mo ago→

Member of Technical Staff, Machine Learning

BJAK·United States·On-site

Staff·ML Engineer

Python PyTorch TensorFlow Natural language processing

2mo ago→

Senior Machine Learning Engineer

BJAK·China·On-site

Senior·ML Engineer

Python PyTorch Natural language processing MLOps

2mo ago→

Senior Machine Learning Engineer

BJAK·United States·On-site

Senior·ML Engineer

Python PyTorch Natural language processing MLOps

2mo ago→

Research Scientist, Simulation

Dyna Robotics·Redwood City, CA·On-site

Senior·Research Scientist

Simulation Robotics Reinforcement learning Procedural generation

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

2mo ago→

Applied Scientist

Mistral AI·Palo Alto, New York·Hybrid

Senior·Research Scientist

Transformer architectures Multimodal ai State-of-the-art model development Fine-tuning

2mo ago→

Applied Scientist - All Levels

PhysicsX·Singapore ·On-site

Research Scientist

Multi-physics simulation Ai-driven inference Foundation models Numerical physics

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

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

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

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

2mo ago→

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