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.

PhD Machine Learning Software Engineer Intern (Summer 2027)
LyftSan Francisco, CA +1
- Intern
- ML / AI engineering
- Python
- SQL
- PyTorch
- Spark
- A/B testing
- +2 more
$65–$68 / hr
Posted Oct 6, 2026
S
Senior HPC & GPU Infrastructure Engineer
SciforiumSan Francisco, CA
- Senior
- Onsite
- Inference & ML systems
- 5+ years
- Python
- Kubernetes
- Terraform
- PyTorch
- Ansible
- +5 more
$150K–$220K / yr
Posted Oct 6, 2026
Senior Applied Scientist, AWS Neuron Science - Core Algorithm

Annapurna Labs (U.S.) Inc.Cupertino, California

Senior
Research
6+ years

Python
AWS
Java
C++
PyTorch
+6 more

$192K–$260K / yr

Posted Oct 6, 2026

checked 2 days ago

IO

Machine Learning Engineer — Reinforcement Learning

Institute of Foundation ModelsSunnyvale, CA

Onsite
ML / AI engineering
5+ years

Python
Kubernetes
PyTorch
CUDA
JAX
+2 more

$150K–$450K / yr

Posted Oct 5, 2026

IO

Machine Learning Engineer — Pre-training (LLM)

Institute of Foundation ModelsSunnyvale, CA

Onsite
Inference & ML systems
5+ years

Python
Kubernetes
PyTorch
CUDA
JAX
+1 more

$150K–$450K / yr

Posted Oct 5, 2026

O

Machine Learning Engineer

OpenMindSan Francisco

Onsite
ML / AI engineering

Python
Go
C++
PyTorch
JAX
+1 more

$130K–$200K / yr

Posted Oct 5, 2026

Senior Software Engineer, Vehicle Interaction

WaymoMountain View, CA +1

Senior
Software engineering
5+ years

C++
PyTorch
JAX

$213K–$263K / yr

Posted Oct 5, 2026

Senior ML Engineer, Computer Vision

Anduril IndustriesIrvine, California +2

Senior
Remote
ML / AI engineering
6+ years

Python
C++
PyTorch
JAX
Linux
+1 more

$220K–$292K / yr

Posted Oct 5, 2026

Machine Learning Researcher - Agentic Science

PostEraRemote (US)

Remote
Research

Python
PyTorch
JAX

$200K–$300K / yr

Posted Oct 5, 2026

checked 2 days ago

Principal, Data Scientist - Applied Machine Learning

Walmart(USA) VIZIO SAN FRANCISCO CA San Francisco Home Office +1

Principal
ML / AI engineering
3+ years

Python
PyTorch
Spark
JAX
TensorFlow

$132K–$286K / yr

Posted Oct 5, 2026

checked 2 days ago

MS

Staff Software Engineer, IR Data Products

Muon SpaceDenver, CO +1

Staff
Remote
Software engineering
10+ years

AWS
Terraform
Airflow
CUDA
JAX
+1 more

$216K–$239K / yr

Posted Oct 2, 2026

SS

Staff Engineer, ML Profiling Tools

Samsung SemiconductorSan Jose, California +1

Staff
ML / AI engineering
10+ years

PyTorch
JAX
TensorFlow

$163K–$253K / yr

Posted Oct 2, 2026

2027 Summer Intern, PhD, Research, World Action Model

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted Oct 2, 2026

CR

Scientist - AI & Machine Learning

Charles River AnalyticsCambridge, MA

ML / AI engineering
3+ years

Python
Docker
PyTorch
JAX
Linux
+1 more

$120K–$200K / yr

Posted Oct 2, 2026

Research Platform Engineer

Tower Research CapitalNew York

Research

Kubernetes
PyTorch
JAX

$200K–$300K / yr

Posted Oct 2, 2026

2027 Summer Intern - AI/ML Engineer, Autonomous Vehicle: Simulation

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
Java
C++
PyTorch
JAX
+1 more

$8K–$10K / mo

Posted Oct 1, 2026

checked 2 days ago

2027 Summer Intern – AI/ML Engineer, Autonomous Vehicles: Simulation (PhD)

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+1 more

$12K–$15K / mo

Posted Oct 1, 2026

checked 2 days ago

AI Lab - Research

Optiver New York, New York +1

Research

Python
PyTorch
CUDA
JAX

$200K / yr

Posted Sep 30, 2026

Software Engineer, Evaluation Infrastructure

xAIPalo Alto, CA

Infra, platform & SRE

Python
C++
Rust
PyTorch
JAX

$180K–$440K / yr

Posted Sep 30, 2026

CP

Research Scientist - RF Machine Learning

Careers PeratonCollege Park, MD

Hardware and robotics
10+ years

Python
Java
C++
PyTorch
C
+9 more

$146K–$234K / yr

Posted Sep 30, 2026

checked 2 days ago

Senior ML Compiler Engineer, Neuron

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

Senior
Inference & ML systems
5+ years

PyTorch
JAX

$168K–$262K / yr

Posted Sep 30, 2026

checked 2 days ago

Research Engineer / Research Scientist, RL Frontiers

AnthropicSan Francisco, CA +2

Research
Visa sponsorship in the job text

Python
C++
Rust
PyTorch
JAX

$500K–$850K / yr

Posted Sep 29, 2026

2027 Summer Intern, PhD, Planner Machine Learning

WaymoSan Francisco, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted Sep 29, 2026

(Jr) Machine Learning (ML) Developer

LaminarBoston, MA

Entry level
Hybrid
ML / AI engineering

Python
AWS
GCP
PyTorch
Databricks
+5 more

$89K–$141K / yr

Posted Sep 29, 2026

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

WaymoMountain View, CA +1

Intern
Research

Python
C++
PyTorch
C
JAX
+1 more

$85 / hr

Posted Sep 28, 2026

TA

Research Engineer, Climate Modeling

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research
2+ years

Python
GCP
JAX

$147K–$220K / yr

Posted Sep 25, 2026

2027 Summer Intern, MS/PhD, Perception, Machine Learning

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 25, 2026

TA

Research Scientist, Climate Modeling

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research

Python
JAX

$167K–$261K / yr

Posted Sep 25, 2026

2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 25, 2026

Senior Director / Vice President, Machine Learning (Generative Chemistry)

Flagship Pioneering, Inc.Cambridge, MA USA +1

Director+
ML / AI engineering
Manager
7+ years

Python
AWS
Docker
Terraform
PyTorch
+7 more

$279K–$413K / yr

Posted Sep 25, 2026

2027 Summer Intern, PhD, Research Scientist, Foundation Models

WaymoMountain View, CA +1

Intern
Research

Python
JAX
TensorFlow

$85 / hr

Posted Sep 24, 2026

2027 Summer Intern, MS/PhD, Machine Learning Engineer

WaymoSan Francisco, CA +1

Intern
ML / AI engineering

Python
C++
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 22, 2026

Staff ML Engineer, VLA Training

Nidus TechnologiesNew York City

Staff
Onsite
ML / AI engineering
5+ years

PyTorch
JAX
ROS

$250K–$350K / yr

Posted Sep 22, 2026

Staff Software Engineer, Machine Learning (Consumer Revenue)

DiscordSan Francisco Bay Area +1

Staff
ML / AI engineering
8+ years

Python
PyTorch
Spark
Flink
JAX
+2 more

$272K–$340K / yr

Posted Sep 22, 2026

Research Scientist, Contextual AI and Multimodal Agents

MetaRedmond, Washington

Onsite
Research
12+ years

Python
PyTorch
JAX

$219K–$301K / yr

Posted Sep 22, 2026

checked 9 days ago

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Annapurna Labs (U.S.) Inc.Cupertino, California

Senior
Inference & ML systems

AWS
PyTorch
JAX

$193K–$262K / yr

Posted Sep 22, 2026

checked 2 days ago

Sr. Software Engineer- AI/ML, Amazon Neuron Training

Annapurna Labs (U.S.) Inc.Cupertino, California

Senior
ML / AI engineering

PyTorch
JAX

$193K–$262K / yr

Posted Sep 22, 2026

checked 2 days ago

2027 Summer Intern, MS/PhD, Software Engineer, Simulation Evaluation ML Model

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
C++
PyTorch
JAX
Linux
+1 more

$70–$85 / hr

Posted Sep 21, 2026

Research, Mid Training

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 21, 2026

Senior Research Scientist, Magnets

SandboxAQUnited States

Senior
Remote
Research
3+ years

Python
PyTorch
JAX

$134K–$252K / yr

Posted Sep 21, 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