Search tech jobs | JobsWithPay

Jobs Ask AI Pay Interviews Companies CompareSign in

The jobs load…

We read every job from the company's own job board. You apply on the company site.Jobs Job types Internships New graduate jobs Companies Pay Reported pay Check an offer Add your pay Interview questions Add your interview questions Compare with LinkedIn Job sites compared Pay transparency laws API for developers Monthly report Companies we cover Make a portfolio page For companies: find people About Privacy

Examples: Remote senior ML jobs that pay $200K or more New graduate software jobs in New York with visa sponsorship Data engineer jobs in Bengaluru that use Spark

Filters (2)

Job typeSoftware engineering46ML / AI engineering491Inference & ML systems366Research188Infra, platform & SRE8Data science11Security engineering1Product management2Quant and trading2Solutions engineering7Hardware and robotics4

LocationBay Area365New York65Remote (US)39Other places in the US97Outside the US52India3Bengaluru1

LevelIntern50Entry level9Senior93Staff54Principal11Director+7

Work typeOnsite128Hybrid50Remote38

Minimum base payNo minimum$100K+429$150K+425$200K+403$250K+324$300K+210

Date postedAll datesLast 24 hours0Last 7 days21Last 30 days132

Years of experience0 to 2 years3 to 5 years6 to 9 years10 years or more

SkillsJAXPythonAWSSQLTypeScriptKubernetesGCPAzureReactDockerJavaGoC++RustTerraformPyTorchSparkPostgreSQLJavaScriptNode.jsSwiftFigmaExcel

Other filtersOnly AI and ML jobsPosted payVisa sponsorship in the job textMember of Technical StaffOnly manager jobsHide jobs that we found on LinkedIn

506 jobs from 7,835 companies, checked on their job boards 5 h ago

Sort

Posted pay×JAX×Remove all filters

Email me new jobs like this

Tech jobs. Only jobs with posted pay. Skills: JAX.

Your email address

Website

One email each day, only when there are new jobs. No account is necessary. You can stop the emails with one click.

GE
Graduate Software Engineer, AI/ ML Test Systems 2027
Graphcore Early CareersBristol, UK +1
- Entry level
- ML / AI engineering
- Python
- C++
- PyTorch
- JAX
- Linux
- +1 more
£45K / yr
Posted 26 days ago
Summer 2027 Intern - AI Research
SalesforceCalifornia - Palo Alto
- Intern
- Research
- Python
- PyTorch
- JAX
$49–$68 / hr
Posted 26 days ago
checked 2 days ago
Human-Robot Interaction Applied Scientist , Fauna
Amazon.com Services LLC

New York, New York

Research
6+ years

Python
Java
C++
PyTorch
Spark
+5 more

$184K–$249K / yr

Posted 26 days ago

checked 2 days ago

Research, Tinker, RL Systems

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow
vLLM

$350K–$475K / yr

Posted 26 days ago

LLM Researcher

NoriSan Francisco, California

Research

Python
PyTorch
JAX

$150K–$250K / yr

Posted 26 days ago

checked 2 days ago

2027 Summer Intern, PhD, Research, VLM Harnesses for AV Planning

WaymoMountain View, CA +1

Intern
Research
3+ years

PyTorch
JAX

$85 / hr

Posted 26 days ago

2027 Summer Intern, PhD, Machine Learning - Simulation & Closed-Loop Training

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted 26 days ago

Senior Research Scientist, Sensor Technology

GoogleMountain View, California

Senior
Onsite
Research
4+ years

PyTorch
JAX
TensorFlow

$207K–$300K / yr

Posted 26 days ago

checked 9 days ago

2027 Summer Intern, PhD, Research Scientist, Visual Reasoning

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted 26 days ago

ML Engineer, Motion Planning

DiDi LabsSan Jose, CA +1

ML / AI engineering
5+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$170K–$339K / yr

Posted 27 days ago

ML Compiler Engineer, Neuron Kernel Interface

Annapurna Labs (U.S.) Inc.Seattle, Washington +1

Inference & ML systems

PyTorch
JAX

$144K–$224K / yr

Posted 29 days ago

checked 2 days ago

Scientist, Discovery Cheminformatics

Octant BioEmeryville, California +1

Data science
2+ years

Python
PyTorch
JAX
NumPy
pandas
+1 more

$145K–$185K / yr

Posted 29 days ago

G

Research Scientist – Diffusion Models

GranicaBay Area Office

Onsite
Research

Python
PyTorch
JAX

$160K–$240K / yr

Posted Oct 15, 2026

Senior Staff Software Engineer, Storage

CrusoeSunnyvale, CA - US

Staff
Onsite
Infra, platform & SRE
12+ years

AWS
GCP
Azure
Go
C++
+8 more

$245K–$290K / yr

Posted Oct 15, 2026

M

Research Scientist – Multimodal Foundation Models

MeckaToronto GTA +2

Hybrid
Research

Python
PyTorch
JAX

$185K–$210K / yr

Posted Oct 15, 2026

A

Senior Perception Learning Engineer

ApptronikSunnyvale, CA +1

Senior
ML / AI engineering
3+ years

Python
C++
PyTorch
JAX
ROS
+1 more

$220K–$250K / yr

Posted Oct 15, 2026

DD

System Software Engineer - AI

Delos Data IncMontreal

Hybrid
Software engineering
5+ years

Python
C++
PyTorch
CUDA
JAX
+2 more

CA$120K–CA$180K / yr

Posted Oct 14, 2026

Senior Computational Scientist, Electric Grid Simulations

XMountain View, CA (HQ)

Senior
Data science
6+ years

Python
Java
C++
C
JAX
+1 more

$174K–$261K / yr

Posted Oct 14, 2026

Research Engineer

TrainLoopSan Francisco, California

Research
3+ years

PyTorch
JAX

$150K–$250K / yr

Posted Oct 14, 2026

checked 2 days ago

H

Autonomy Engineer - VLA Pre-training

HumanoidUS, San Diego, CA

Onsite
Inference & ML systems
3+ years

Python
PyTorch
JAX

$180K–$300K / yr

Posted Oct 14, 2026

Software Engineer, ML Accelerators, DeepMind

GoogleMountain View, California +1

Onsite
Inference & ML systems

Python
C++
PyTorch
JAX

$174K–$252K / yr

Posted Oct 14, 2026

checked 9 days ago

Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)

CohereEurope +2

Remote
Inference & ML systems
Member of Technical Staff

PyTorch
JAX

CA$295K–CA$535K / yr

Posted Oct 14, 2026

Software Engineer, Systems ML - Compilers / Kernels

MetaMenlo Park, California +3

Onsite
Inference & ML systems
6+ years

PyTorch
A/B testing
CUDA
JAX
TensorFlow

$154K–$217K / yr

Posted Oct 14, 2026

checked 9 days ago

Principal Engineer: Annapurna Labs Training and Inference , Annapurna Labs, AWS Neuron SDK

Annapurna Labs (U.S.) Inc.Seattle, Washington +1

Principal
Inference & ML systems

AWS
JAX

$200K–$311K / yr

Posted Oct 14, 2026

checked 2 days ago

2027 Summer Intern, MS/PhD, Machine Learning, Planning/Prediction

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
C++
PyTorch
C
JAX
+1 more

$70–$85 / hr

Posted Oct 13, 2026

R

Scientist, Plume Modeling and Forecasting

ReflectiveRemote or San Francisco, California +2

Remote
Data science
Visa sponsorship in the job text

Python
JAX

$130K–$160K / yr

Posted Oct 13, 2026

2027 Summer Intern, PhD, Machine Learning, Simulation Realism

WaymoLondon, England, United Kingdom +1

Intern
ML / AI engineering

Python
PyTorch
JAX
NumPy
TensorFlow

£80K / yr

Posted Oct 13, 2026

2027 Summer Intern, MS/PhD, Perception, Model Lifecycle

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Oct 12, 2026

H

Member of Technical Staff, LLM & Production

HarkSan Jose +1

ML / AI engineering
Member of Technical Staff
3+ years

Python
PyTorch
JAX
vLLM

$180K–$450K / yr

Posted Oct 12, 2026

2027 Summer Intern, PhD, Research, AV Planning

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX

$85 / hr

Posted Oct 9, 2026

H

Director of Autonomy - Biped

HumanoidUS, San Diego, CA

Director+
Onsite
ML / AI engineering
Manager
5+ years

PyTorch
JAX

$300K–$450K / yr

Posted Oct 9, 2026

Senior Software Engineer, Generative AI, Google Research

GoogleSunnyvale, California

Senior
Onsite
Research

Python
C++
PyTorch
JAX

$174K–$252K / yr

Posted Oct 9, 2026

checked 9 days ago

Research Engineer, Agentic Security for Gemini, DeepMind

GoogleMountain View, California +1

Onsite
Research
1+ years

Python
PyTorch
JAX

$174K–$252K / yr

Posted Oct 8, 2026

checked 9 days ago

2027 Summer Intern, PhD, Machine Learning, Planning/Prediction

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted Oct 8, 2026

GA

Research Scientist Intern, AI Molecular Design (Winter/Spring)

GenBio AIPalo Alto, CA

Intern
Onsite
Research

PyTorch
JAX
TensorFlow

$8K–$10K / mo

Posted Oct 8, 2026

Director of Product Management, Agentforce Voice Models

SalesforceCalifornia - San Francisco +1

Director+
Product management
Manager
10+ years

Python
AWS
GCP
Azure
PyTorch
+1 more

$197K–$345K / yr

Posted Oct 8, 2026

checked 2 days ago

T

Machine Learning Engineer, Applied

TraceUnited States +2

Remote
ML / AI engineering
2+ years

Python
PyTorch
JAX

$185K–$245K / yr

Posted Oct 7, 2026

CI

Software Engineer, Numerical Computing

CHAOS IndustriesWashington, District of Columbia +1

Software engineering
5+ years

Python
PyTorch
CUDA
JAX
Jira
+2 more

$150K–$210K / yr

Posted Oct 7, 2026

TU

AI Software Intern (USA)

Tenstorrent University JobsAustin, Texas +3

Intern
Software engineering

Python
C++
PyTorch
C
CUDA
+2 more

$50–$70 / hr

Posted Oct 7, 2026

2027 Summer Intern, MS/PhD, Research, Perception Foundation Models

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Oct 7, 2026

Go to the first page Show the next 40 jobs

JobsWithPay agent

Sign in to use it

Free in beta

Sign in to use the agent

The agent changes this search while you chat. It also compares pay, compares jobs with your résumé and saves jobs for you. All features are free while JobsWithPay is in beta.

Sign in for free