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

Machine Learning Engineer
Hestus, Inc.San Mateo, California
- ML / AI engineering
- 4+ years
- Python
- AWS
- SQL
- GCP
- Azure
- +5 more
$180K–$240K / yr
Posted Mar 9, 2026
checked 2 days ago
M
Expression of Interest: Machine Learning Engineer
MolocoMenlo Park, California +5
- ML / AI engineering
- 5+ years
- Python
- AWS
- Kubernetes
- GCP
- Go
- +6 more
$170K–$386K / yr
Posted Mar 6, 2026
Systems Engineer
General RoboticsRedmond, WA

Infra, platform & SRE
1+ years

Python
AWS
Kubernetes
GCP
Azure
+8 more

$155K–$205K / yr

Posted Feb 25, 2026

Research Scientist: Hierarchical Sensorimotor Perception

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

Research Scientist: Hierarchical RL Agents

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

Research Scientist/Engineer

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

Research Scientist: Energy Based Models

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

Staff Data Scientist | NLP

MachinifyRemote - US

Staff
Remote
ML / AI engineering
3+ years

Python
AWS
SQL
GCP
Azure
+4 more

$180K–$230K / yr

Posted Feb 24, 2026

CP

AI Solutions Architect | US

Cuesta PartnersRemote US

Remote
Solutions engineering
8+ years

AWS
Kubernetes
GCP
Azure
Docker
+12 more

$145K–$188K / yr

Posted Feb 22, 2026

Senior Scientist, Machine Learning

Altos LabsSan Francisco Bay Area, CA +2

Senior
ML / AI engineering
5+ years

Python
PyTorch
Hugging Face
JAX

$179K–$330K / yr

Posted Feb 17, 2026

Research Engineer, Machine Learning (Reinforcement Learning)

AnthropicLondon, UK

Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

£260K–£630K / yr

Posted Feb 11, 2026

Machine Learning Performance Engineer

Optiver New York, New York +1

ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+2 more

$200K / yr

Posted Feb 5, 2026

Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

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

Inference & ML systems

AWS
PyTorch
JAX

$144K–$224K / yr

Posted Feb 5, 2026

checked 2 days ago

ET

Principal Solutions Architect (Req #1048)

ePlus Technology, inc.San Ramon, CA +1

Principal
Solutions engineering
8+ years

AWS
Kubernetes
GCP
Azure
PyTorch
+8 more

$170K–$190K / yr

Posted Feb 4, 2026

Software Development Manager - Compiler

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

Inference & ML systems
Manager

AWS
PyTorch
JAX

$213K–$288K / yr

Posted Feb 4, 2026

checked 2 days ago

Quantitative Developer (Python) - Central Liquidity Strategies

MillenniumNew York, New York

Onsite
Quant and trading
5+ years

Python
C++
Rust
PyTorch
CUDA
+2 more

$160K–$250K / yr

Posted Feb 3, 2026

checked 2 days ago

Member of Technical Staff - Quantitative Research

UdioNew York City (Remote possible for exceptional candidates) +1

Remote
Quant and trading
Member of Technical Staff
5+ years

Kubernetes
GCP
JAX
TensorFlow

$250K–$350K / yr

Posted Jan 22, 2026

D

Reinforcement Learning Engineer

DexmateFremont Office

Onsite
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$120K–$300K / yr

Posted Jan 19, 2026

Head of Engineering, Platform

GoodfireSan Francisco, CA +1

Director+
Infra, platform & SRE
Manager
10+ years

PyTorch
JAX

$300K–$450K / yr

Posted Jan 15, 2026

Member of Technical Staff, Research Engineer

RunwayRemote

Research
Member of Technical Staff
4+ years

PyTorch
JAX

$270K–$370K / yr

Posted Jan 9, 2026

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQUnited States

Staff
Remote
ML / AI engineering
8+ years

Python
AWS
GCP
PyTorch
JAX
+4 more

$157K–$294K / yr

Posted Jan 9, 2026

Staff Product Manager, Training Infrastructure - Weights & Biases

CoreWeaveLivingston, NJ / New York, NY / San Francisco, CA / Sunnyvale, CA / Bellevue, WA +1

Staff
Product management
7+ years

Kubernetes
PyTorch
JAX

$188K–$275K / yr

Posted Jan 8, 2026

AI Applied Scientist

FigmaSan Francisco, CA • New York, NY • United States +2

Research
6+ years

Python
Java
C++
PyTorch
Figma
+3 more

$153K–$376K / yr

Posted Dec 17, 2025

SE

Senior Program Scientist, AI and Advanced Computing Institute

Schmidt EntitiesNew York, NY

Senior
Onsite
Data science
10+ years

Python
PyTorch
JAX

$300K–$325K / yr

Posted Dec 16, 2025

Member of Technical Staff, Data Analysis and Evaluation

CohereLondon +5

Remote
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

CA$250K–CA$535K / yr

Posted Dec 15, 2025

Senior Machine Learning Engineer, Driver Refinement & Validation

WaymoMountain View, CA +5

Senior
ML / AI engineering
5+ years

Python
C++
PyTorch
JAX

$213K–$263K / yr

Posted Dec 15, 2025

D

Robot Learning Engineer

DexmateFremont Office

Onsite
Software engineering
3+ years

Python
PyTorch
JAX
TensorFlow

$100K–$300K / yr

Posted Dec 15, 2025

R

Senior / Staff AI Research Engineer, Data Infrastructure

RoboForceMilpitas, CA

Staff
Research
5+ years
Visa sponsorship in the job text

Python
PyTorch
JAX
MLflow

$185K–$325K / yr

Posted Dec 11, 2025

X

Member of Technical Staff, Robotics

XDOFSan Mateo On-site

Onsite
ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX

$175K–$275K / yr

Posted Dec 10, 2025

Staff Machine Learning Engineer, Mapping

WaymoMountain View, CA +1

Staff
ML / AI engineering
8+ years

PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted Dec 9, 2025

Staff Tech Lead, Machine Learning Engineer, Perception

WaymoMountain View, California +3

Staff
ML / AI engineering
7+ years

Python
C++
PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted Dec 9, 2025

Staff Machine Learning Engineer, Infrastructure

WaymoMountain View, CA +3

Staff
ML / AI engineering
6+ years

PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted Dec 8, 2025

Senior Staff ML Engineer, Driver Understanding and Evaluation

WaymoMountain View, CA +1

Staff
ML / AI engineering
10+ years

Python
JAX
TensorFlow

$281K–$356K / yr

Posted Dec 5, 2025

Sr. Machine Learning Engineer, Ads

xAIPalo Alto, CA

Senior
ML / AI engineering
5+ years

PyTorch
ClickHouse
CUDA
JAX
Kafka

$180K–$330K / yr

Posted Dec 3, 2025

Deep Learning Researcher

IMCNew York +1

Research
3+ years

Python
PyTorch
JAX
TensorFlow

$250K–$300K / yr

Posted Nov 28, 2025

Senior AI Performance Engineer

SambaNovaSan Jose, California +1

Senior
Software engineering
3+ years

Python
C++
PyTorch
CUDA
JAX
+2 more

$180K–$255K / yr

Posted Nov 21, 2025

TT

Senior Machine Learning Scientist

Tahoe TherapeuticsSouth San Francisco

Senior
Onsite
ML / AI engineering

PyTorch
JAX
NumPy
pandas
TensorFlow

$200K–$275K / yr

Posted Nov 17, 2025

S

Senior Research Scientist

SciforiumSan Francisco, CA

Senior
Onsite
Research
5+ years

Python
PyTorch
JAX

$190K–$250K / yr

Posted Nov 15, 2025

TR

Human Interactive Driving Research Intern, World Models

Toyota Research InstituteLos Altos, CA

Intern
Hybrid
Research
2 positions

AWS
PyTorch
JAX
TensorFlow

$45–$65 / hr

Posted Nov 14, 2025

Member of Technical Staff, Quantitative Systems

Autonomous Technologies GroupSan Francisco, California +1

Inference & ML systems
Member of Technical Staff

Python
Go
C++
Rust
PyTorch
+3 more

$200K–$300K / yr

Posted Nov 14, 2025

checked 2 days ago

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