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

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

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.

RA
Research Member of Technical Staff- Training Systems
Rhoda AIMountain View
- Research
- Member of Technical Staff
- PyTorch
- CUDA
- JAX
$200K–$300K / yr
Posted Sep 18, 2026
Research Intern, Frontier Agents (Summer 2027)
Together AISan Francisco
- Intern
- Research
- Python
- PyTorch
- JAX
- Salesforce
$58–$70 / hr
Posted Sep 18, 2026
S
Member of Technical Staff, Data and RL
SonderNew York City
- Onsite
- ML / AI engineering
- Member of Technical Staff
- Visa sponsorship in the job text

Python
PyTorch
JAX

$250K–$300K / yr

Posted Sep 18, 2026

Machine Learning Engineer, Driver Refinement & Validation, Perception

WaymoMountain View, CA +3

ML / AI engineering
3+ years

Python
C++
PyTorch
JAX

$175K–$215K / yr

Posted Sep 18, 2026

Research Intern, Model Shaping (Summer 2027)

Together AISan Francisco, Amsterdam +1

Intern
Research

PyTorch
JAX
Salesforce

$58–$70 / hr

Posted Sep 18, 2026

Research Intern, Inference (Summer 2027)

Together AISan Francisco

Intern
Research

Python
PyTorch
CUDA
JAX
Salesforce

$58–$70 / hr

Posted Sep 18, 2026

Senior Computer Vision Engineer (Localization)

Niantic SpatialSan Francisco, CA +1

Senior
Hybrid
ML / AI engineering
5+ years

Python
C++
PyTorch
JAX

$189K–$255K / yr

Posted Sep 18, 2026

Research Engineer

AdobeSeattle

Research
3+ years

Python
C++
PyTorch
JAX
TensorFlow

$121K–$239K / yr

Posted Sep 18, 2026

checked 2 days ago

Partner Engineering GenAI - US

MetaMenlo Park, California

Onsite
Software engineering
5+ years

Python
Java
Go
C++
Rust
+4 more

$212K–$250K / yr

Posted Sep 18, 2026

checked 9 days ago

Staff Tech Lead Manager, Machine Learning Engineering, Simulator Evaluation

WaymoMountain View, California +2

Staff
ML / AI engineering
Manager
7+ years

Python
C++
PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted Sep 17, 2026

Member of Technical Staff, AI Agent Development Lead

PostmanSan Francisco, CA

Senior
ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

$256K–$276K / yr

Posted Sep 17, 2026

checked 2 days ago

Research, Finetuning Science

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 16, 2026

Research Lead, Tinker, Fine-tuning Science

Thinking Machines LabSan Francisco

Senior
Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 16, 2026

2027 Summer Intern, PhD, Machine Learning, Computer Vision

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted Sep 16, 2026

AA

GPU Kernel Engineer – CUDA, Triton & Accelerator Performance

Anyone AIArgentina - Fully Remote +8

Remote
Inference & ML systems
3+ years

AWS
CUDA
JAX

$65 / hr

Posted Sep 15, 2026

Cybersecurity Engineer

VisaUS - Foster City, CA

Security engineering
5+ years

Python
AWS
SQL
Kubernetes
GCP
+12 more

$137K–$213K / yr

Posted Sep 15, 2026

checked 2 days ago

Machine Learning Engineer – Ad Sciences

InMobi San Mateo, CA +1

ML / AI engineering
3+ years

JAX

$173K–$210K / yr

Posted Sep 14, 2026

Worldwide Specialist Solutions Architect - GenAI, Data & AI GTM

Amazon Web Services, Inc.Austin, Texas +3

Solutions engineering

AWS
Kubernetes
PyTorch
JAX

$154K–$239K / yr

Posted Sep 14, 2026

checked 2 days ago

RA

Software Engineer, Distributed Training

River AI Inc.Palo Alto, CA +1

Inference & ML systems
Visa sponsorship in the job text

Python
C++
Rust
PyTorch
CUDA
+1 more

$200K–$420K / yr

Posted Sep 12, 2026

Member of Technical Staff - Machine Learning

Inductive BioNew York City, San Francisco, or Boston

Hybrid
ML / AI engineering
Member of Technical Staff
4+ years

PyTorch
JAX
TensorFlow

$180K–$250K / yr

Posted Sep 11, 2026

Senior Machine Learning Engineer, Predictive Planner, Systems Arch

WaymoMountain View, CA +3

Senior
ML / AI engineering
4+ years

Python
C++
PyTorch
C
JAX
+1 more

$213K–$263K / yr

Posted Sep 10, 2026

2027 Summer Intern, PhD, Vehicle Intent and Prediction

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted Sep 10, 2026

Staff+ Software Engineer, ML Inference Path

AnthropicSan Francisco, CA

Staff
Inference & ML systems
5+ years
Visa sponsorship in the job text

Python
PyTorch
A/B testing
JAX
TensorFlow

$320K–$485K / yr

Posted Sep 9, 2026

TA

Machine Learning Engineer (II-III), Space Edge Deployment

True AnomalyDenver, CO +1

ML / AI engineering
2+ years

Python
PyTorch
JAX
TensorFlow

$125K–$220K / yr

Posted Sep 9, 2026

Research Software Engineer, Post Training

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 9, 2026

E

Thermo ML Resident

ExtropicSan Francisco +1

Entry level
Onsite
ML / AI engineering
2 positions

Python
Kubernetes
PyTorch
JAX
Ray
+1 more

$75K–$200K / yr

Posted Sep 8, 2026

Research Scientist, Map Scalability

WaymoMountain View, CA, U.S.A +5

Research
1+ years

Python
C++
PyTorch
JAX
TensorFlow

$213K–$263K / yr

Posted Sep 8, 2026

FP

Pioneering Intelligence: Data Science Co-Op (Embedded Science Team)

Flagship Pioneering Co-Op ProgramCambridge, MA USA +1

Intern
Data science

Python
AWS
GCP
Azure
PyTorch
+1 more

$25–$35 / hr

Posted Sep 8, 2026

Software Engineer, GenAI Frameworks

MetaMenlo Park, California

Onsite
Software engineering
2+ years

Python
Java
C++
PyTorch
Spark
+4 more

$122K–$181K / yr

Posted Sep 8, 2026

checked 9 days ago

Founding Research Scientist, Biological Foundation Models

CellTypeNew York, New York

Research

Python
PyTorch
JAX

$150K–$250K / yr

Posted Sep 7, 2026

checked 2 days ago

Robotics Autonomy Engineer, Manipulation

FieldAIIrvine, CA +1

Onsite
ML / AI engineering
3+ years

Python
C++
PyTorch
JAX
ROS
+1 more

$100K–$200K / yr

Posted Sep 5, 2026

Applied Scientist, Neuron ARG, Annapurna ML

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

Research
4+ years

Python
Java
C++
PyTorch
JAX

$143K–$222K / yr

Posted Sep 5, 2026

checked 2 days ago

Computational Scientist, Differentiable Physics

Periodic LabsMenlo Park, CA

Onsite
Data science
Visa sponsorship in the job text

PyTorch
JAX

$250K–$350K / yr

Posted Sep 4, 2026

2027 PhD Residency, Earth Scientist AI Resident (Early Stage Project)

XMountain View, CA +1

Entry level
Data science

Python
JAX
NumPy
scikit-learn
TensorFlow

$109K–$157K / yr

Posted Sep 4, 2026

Staff Tech Lead, Perception, Pre-training & Distillation

WaymoMountain View, California, United States, San Francisco, California +2

Staff
Inference & ML systems
5+ years

Python
PyTorch
JAX

$251K–$310K / yr

Posted Sep 1, 2026

Sr Staff Tech Lead, Machine Learning Engineer, Perception

WaymoMountain View, California, United States, San Francisco, California +2

Staff
ML / AI engineering
8+ years

Python
C++
PyTorch
JAX
TensorFlow

$298K–$368K / yr

Posted Sep 1, 2026

Senior Machine Learning Engineer, Vehicle Interaction & Behavior Policy ML

WaymoMountain View, California +1

Senior
ML / AI engineering
2+ years

PyTorch
JAX

$213K–$263K / yr

Posted Aug 31, 2026

Senior Machine Learning Engineer, Multimodal Perception

WaymoMountain View, California +1

Senior
ML / AI engineering
2+ years

PyTorch
JAX

$213K–$263K / yr

Posted Aug 31, 2026

S

Pre-training Research Engineer

SciforiumSan Francisco, CA

Onsite
Research
5+ years

JAX

$165K–$225K / yr

Posted Aug 31, 2026

S

Research Engineer - Model Evaluation & MLOps

SciforiumSan Francisco, CA

Onsite
Research
2+ years

Python
PyTorch
Hugging Face
JAX
TensorFlow
+1 more

$155K–$200K / yr

Posted Aug 31, 2026

Go to the first page Show the next 40 jobs