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.

Senior / Staff Machine Learning Research Engineer
Calico South San Francisco, CA
- Staff
- Research
- 7+ years
- Python
- Kubernetes
- GCP
- Docker
- PyTorch
- +1 more
$220K–$290K / yr
Posted May 5, 2026
PM
Member of Technical Staff - Machine Learning Capabilities, New Graduates
Preference ModelSan Francisco
- Entry level
- Onsite
- ML / AI engineering
- Member of Technical Staff
- Visa sponsorship in the job text
- Python
- AWS
- GCP
- Azure
- PyTorch
- +3 more
$165K–$200K / yr
Posted May 4, 2026

Performance Engineer (Inference, Training & GPU)

World LabsSan Francisco

Onsite
Inference & ML systems
Visa sponsorship in the job text

Python
Go
C++
Rust
PyTorch
+2 more

$200K–$300K / yr

Posted May 1, 2026

A

Machine Learning Research Engineer (MLRE) - GPUs

AchiraSan Francisco Office +2

Hybrid
Research
2+ years

PyTorch
CUDA
JAX

$165K–$259K / yr

Posted Apr 29, 2026

A

Machine Learning Research Engineer (MLRE) - Workflows/Systems

AchiraSan Francisco Office +1

Hybrid
Research
2+ years

PyTorch
JAX

$165K–$259K / yr

Posted Apr 29, 2026

A

Machine Learning Research Engineer (MLRE) - Research

AchiraSan Francisco Office +1

Hybrid
Research

PyTorch
JAX

$165K–$259K / yr

Posted Apr 29, 2026

Senior Staff Machine Learning Engineer - Content Platform

SpotifyNew York, NY

Staff
Remote
ML / AI engineering

PyTorch
JAX
TensorFlow

$281K–$402K / yr

Posted Apr 29, 2026

Senior Machine Learning Engineer, Runtime and Serving

WaymoMountain View, CA +1

Senior
Inference & ML systems
5+ years

Python
C++
PyTorch
CUDA
JAX

$213K–$263K / yr

Posted Apr 29, 2026

Model Engineer, Post-Training & RL

xAIPalo Alto, CA

Inference & ML systems

Python
C++
Rust
PyTorch
JAX

$160K–$250K / yr

Posted Apr 29, 2026

TA

Research Engineer, Robotics: Physics Simulation and Sim-to-Real

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research
3+ years

Python
C++
CUDA
JAX

$129K–$193K / yr

Posted Apr 28, 2026

LS

Senior / Principal ML Scientist, Foundation Models for Life Sciences

Lila SciencesSan Francisco, CA USA +1

Principal
ML / AI engineering

AWS
GCP
PyTorch
JAX
TensorFlow

$268K–$384K / yr

Posted Apr 27, 2026

AI

Director, Machine Learning, Virtual Cell Initiative

Arc InstitutePalo Alto, CA +1

Director+
ML / AI engineering
Manager
5+ years

PyTorch
JAX
TensorFlow

$380K–$420K / yr

Posted Apr 27, 2026

9M

Perception Engineer

9 MothersAustin

Onsite
ML / AI engineering
6+ years

Python
C++
Rust
PyTorch
JAX

$110K–$300K / yr

Posted Apr 23, 2026

Research Engineer, Machine Learning (RL Velocity)

AnthropicLondon, UK

Research
Visa sponsorship in the job text

PyTorch
JAX

£370K–£630K / yr

Posted Apr 23, 2026

Research Engineer, Machine Learning (RL Velocity)

AnthropicRemote-Friendly (Travel-Required) +2

Remote
Research
Visa sponsorship in the job text

PyTorch
JAX

$500K–$850K / yr

Posted Apr 23, 2026

E

Research Scientist - Machine Learning

ExtropicBoston +1

Onsite
Research
2 positions

Python
AWS
PyTorch
JAX
TensorFlow

$150K–$250K / yr

Posted Apr 22, 2026

F

Research Engineer

ForusNew York office

Research

PyTorch
JAX
TensorFlow

$170K–$300K / yr

Posted Apr 22, 2026

Sr. Software Engineer, Computer Vision

SpaceXHawthorne, CA

Senior
ML / AI engineering
8+ years

Python
Kubernetes
Docker
C++
PyTorch
+11 more

$160K–$225K / yr

Posted Apr 21, 2026

Senior Cluster Site Reliability Engineer

The Voleon GroupBerkeley, CA +1

Senior
Remote
Infra, platform & SRE
5+ years

Python
AWS
GCP
Docker
Terraform
+9 more

$205K–$235K / yr

Posted Apr 17, 2026

Strategic Technical Account Manager GPU

VultrRemote - United States

Remote
Solutions engineering
2+ years

Kubernetes
PyTorch
CUDA
Hugging Face
JAX
+3 more

$115K–$140K / yr

Posted Apr 16, 2026

Forward Deployed Engineer

GoodfireSan Francisco, CA +1

Software engineering

Python
PyTorch
JAX

$200K–$325K / yr

Posted Apr 14, 2026

Research Engineer, Bellwether, X

XMountain View, CA (HQ)

Research
5+ years

Python
PyTorch
JAX
NumPy
pandas
+1 more

$174K–$261K / yr

Posted Apr 13, 2026

Member of Technical Staff (AI Inference Engineer)

PerplexitySan Francisco +2

Inference & ML systems
Member of Technical Staff
3+ years

Python
Rust
PyTorch
CUDA
JAX
+1 more

$220K–$485K / yr

Posted Apr 13, 2026

AI/ML Research Engineer

Manifold BioBoston, MA or San Francisco, CA +2

Research
2+ years

Python
AWS
Kubernetes
GCP
Docker
+5 more

$140K–$225K / yr

Posted Apr 13, 2026

Research Engineer, World Models

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX
Ray

$155K–$269K / yr

Posted Apr 2, 2026

XT

BioMedical AI Research Engineer

Xaira TherapeuticsSouth San Francisco, California +1

Research

Python
PyTorch
JAX
TensorFlow

$252K–$308K / yr

Posted Mar 31, 2026

Systems Engineer - Machine Learning

General RoboticsRedmond, WA

Onsite
ML / AI engineering
1+ years

Python
AWS
Kubernetes
GCP
Azure
+3 more

$155K–$200K / yr

Posted Mar 30, 2026

Machine Learning Researcher - Springtail

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
CUDA
JAX

$150K–$300K / yr

Posted Mar 29, 2026

Research Scientist, World Models

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX

$155K–$269K / yr

Posted Mar 27, 2026

Research Scientist, Neural Reconstruction

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX

$155K–$269K / yr

Posted Mar 27, 2026

Senior / Staff Software Engineer, ML Datasets & Data Pipelines

WaabiToronto, ON +2

Staff
Hybrid
ML / AI engineering
4+ years

Python
AWS
GCP
PyTorch
Spark
+4 more

$148K–$260K / yr

Posted Mar 26, 2026

DU

Senior/Staff Deep Reinforcement Learning Engineer - DoorDash Dot

DoorDash USASan Francisco, CA

Staff
ML / AI engineering

JAX

$168K–$247K / yr

Posted Mar 25, 2026

Senior Applied ML Engineer

CoreWeaveBellevue, WA / Sunnyvale, CA +2

Senior
ML / AI engineering
4+ years

Python
Kubernetes
PyTorch
PostgreSQL
FastAPI
+1 more

$182K–$242K / yr

Posted Mar 25, 2026

VF

Machine Learning Researcher

Virtu FinancialNew York

Research
2+ years

Python
PyTorch
JAX

$200K–$300K / yr

Posted Mar 24, 2026

Senior Member of Technical Staff, Safety and Security for Agents

CohereLondon +5

Senior
Hybrid
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

CA$250K–CA$535K / yr

Posted Mar 23, 2026

Machine Learning Engineer, Senior

9 MothersAustin, Texas

Senior
ML / AI engineering

Python
C++
Rust
PyTorch
JAX

$150K–$250K / yr

Posted Mar 18, 2026

checked 2 days ago

Model Engineer, Voice

xAIPalo Alto, CA

Software engineering

Python
Kubernetes
C++
Rust
PyTorch
+2 more

$160K–$250K / yr

Posted Mar 16, 2026

Tech Lead Manager ML Optimization

WaymoMountain View, California +1

Senior
ML / AI engineering
Manager
10+ years

PyTorch
JAX
TensorFlow

$298K–$378K / yr

Posted Mar 13, 2026

O

Member of Technical Staff, ML Performance

OdysseyPalo Alto +2

Hybrid
Inference & ML systems
Member of Technical Staff
8+ years

PyTorch
JAX

$149K–$462K / yr

Posted Mar 11, 2026

Research Systems Engineer

Applied ComputeSan Francisco

Onsite
Research
Visa sponsorship in the job text

PyTorch
JAX

$190K–$310K / yr

Posted Mar 10, 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