JAX jobs

As of 2026-06-11, ML/AI Work tracks 934 active AI roles using JAX across the US and Europe; the median advertised salary is $161,000–$258,100/yr across 693 roles that disclose pay (ML/AI Work, dataset v2026.06).

Latest JAX jobs

View all →

Matterport – Senior Machine Learning/Computer Vision Engineer – 3D Reconstruction and Semantic Understanding

CoStar Group · Remote · Oakland

$190,000 – $253,000/yrRemoteSenior2 days ago

Research Scientist

— · Oakland, US

$150,000 – $350,000/yr2 days ago

Technical Lead Manager, TorchTPU

Google · London, GB

Lead2 days ago

AI/ML Engineer

Vantor · Remote · Baltimore

$137,000 – $200,200/yrRemote2 days ago

Applied Research Intern, Proactive Intelligence & Customer World Models (PhD / Graduate Co-op)

Block · Remote · San Francisco

RemoteIntern2 days ago

Audio ML Engineer II

Reality Defender · Remote · New York

$160,000 – $210,000/yrRemote2 days ago

Machine Learning Engineer

IMC · Rotterdam, NL

2 days ago

Machine Learning Engineer

IMC · London, GB

2 days ago

Postdoctoral Research Scientist – AI for Bionanoscience

University of Oxford · Coventry, GB

2 days ago

Research Engineer, Pretraining, DeepMind

DeepMind · New York, US

$174,000 – $253,000/yr2 days ago

Senior Research Engineer, On-Device Inference, Robotics, DeepMind

DeepMind · San Jose, US

$207,000 – $301,000/yrSenior2 days ago

Lead Applied Data Scientist - Digital Item RecSys (applied ML, deep learning)(Remote Or Hybrid)

Target · Remote · Minneapolis

$132,000 – $286,000/yrRemoteLead3 days ago

Senior Java Engineer - Focus: Agentic AI

ELCA · Remote · Zurich

RemoteSenior3 days ago

Applied AI Engineer

TU Delft · Rotterdam, NL

3 days ago

Software Development Engineer, AWS

Amazon.com · Baltimore, US

$143,700 – $194,400/yr3 days ago

Software Development Engineer, AWS

Amazon.com · Seattle, US

$143,700 – $194,400/yr3 days ago

Scientist, Structural Chemistry AI/ML

Gilead Sciences · San Jose, US

$146,540 – $189,640/yr3 days ago

Software Development Engineer, AWS

Amazon.com · San Jose, US

$143,700 – $194,400/yr3 days ago

AI Engineer / Data Scientist, AI Manager

PwC · New York, US

$99,000 – $297,000/yr3 days ago

Software Development Engineer, AWS

Amazon.com · New York, US

$143,700 – $194,400/yr3 days ago