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 (1)

Job typeSoftware engineering69ML / AI engineering841Inference & ML systems642Research340Infra, platform & SRE18Data engineering1Data science19Security engineering2Product management2Quant and trading6Solutions engineering14Hardware and robotics4

LocationBay Area527New York85Remote (US)59Other places in the US161Remote (all countries)2Outside the US176India20Bengaluru14Pune3Remote (India)4

LevelIntern103Entry level21Senior135Staff65Principal20Director+11

Work typeOnsite259Hybrid112Remote74

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

Date postedAll datesLast 24 hours0Last 7 days38Last 30 days256

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

869 jobs from 7,835 companies, checked on their job boards 6 h ago

Sort

JAX×Remove all filters

Email me new jobs like this

Tech jobs. 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.

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
PI
ML Infra Engineer, Modeling
Physical IntelligenceSan Francisco

Onsite

ML / AI engineering

AWS
Kubernetes
GCP
PyTorch
JAX

Pay not posted

Posted Sep 1, 2026

Perception Algorithm Engineer

AppleCupertino, California

Onsite
ML / AI engineering
2+ years

Python
C++
PyTorch
Swift
JAX

Pay not posted

Posted Aug 31, 2026

checked 9 days ago

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

EH

Research Engineer / Research Scientist, Vision-Language Model Pretraining

Epsilon HealthSan Francisco, CA

Onsite
Research

PyTorch
JAX

Pay not posted

Posted Aug 31, 2026

EH

Research Engineer / Research Scientist, Vision Foundation Models

Epsilon HealthSan Francisco, CA

Onsite
Research

PyTorch
JAX

Pay not posted

Posted Aug 31, 2026

EH

Research Engineer / Research Scientist, Post-Training & Reinforcement Learning

Epsilon HealthSan Francisco, CA

Onsite
Research

PyTorch
JAX
vLLM

Pay not posted

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

AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - Pre-training Infrastructure

AppleCalifornia

Staff
Onsite
Inference & ML systems
6+ years

Python
PyTorch
CUDA
JAX

Pay not posted

Posted Aug 31, 2026

checked 9 days ago

ML Engineer, Inference Optimization

BuildSan Francisco

Onsite
Inference & ML systems

Python
C++
Rust
PyTorch
CUDA
+1 more

$200K–$320K / yr

Posted Aug 30, 2026

Member of Technical Staff, AI-Driven Compilation

SF TensorSan Francisco

Inference & ML systems
Member of Technical Staff

PyTorch
JAX

$275K–$315K / yr

Posted Aug 29, 2026

checked 2 days ago

Member of Technical Staff, Post-Training & Applied Research

SF TensorSan Francisco

Research
Member of Technical Staff

PyTorch
JAX

$275K–$315K / yr

Posted Aug 29, 2026

checked 2 days ago

SF

Member of Technical Staff, Post-Training & Applied Research

San Francisco Tensor CompanyBay Area

Onsite
Research
Member of Technical Staff

PyTorch
JAX

$275K–$315K / yr

Posted Aug 29, 2026

SF

Member of Technical Staff, AI-Driven Compilation

San Francisco Tensor CompanyBay Area

Onsite
Inference & ML systems
Member of Technical Staff

PyTorch
JAX

$275K–$315K / yr

Posted Aug 29, 2026

AI Security Researcher/Research Engineer

BraveNorth America, EU, UK

Research

Python
PyTorch
JAX

Pay not posted

Posted Aug 29, 2026

Q

Applied Scientist, AI/ML

Hybrid
Research

Python
PyTorch
JAX
pandas
scikit-learn

Pay not posted

Posted Aug 29, 2026

On-device ML Performance Engineer, Graphics, Games and Machine Learning

AppleSeattle, Washington

Onsite
Inference & ML systems

Python
C++
PyTorch
Swift
Bash
+4 more

Pay not posted

Posted Aug 29, 2026

checked 9 days ago

AI/ML Performance Analysis and Validation Engineer

AppleCupertino, California

Onsite
Inference & ML systems
3+ years

Python
PyTorch
JAX

Pay not posted

Posted Aug 28, 2026

checked 9 days ago

MLOps Research Engineer

Merge LabsSan Francisco Bay Area

Onsite
Research

Python
AWS
Kubernetes
GCP
Azure
+4 more

$200K–$265K / yr

Posted Aug 28, 2026

AI Researcher

Hudson River TradingLondon, United Kingdom +3

Research
2+ years

C++
PyTorch
FPGA
JAX

$200K–$300K / yr

Posted Aug 28, 2026

Research Engineer

AfterQuerySan Francisco

Onsite
Research
2+ years

Python
PyTorch
JAX
Ray

$210K–$450K / yr

Posted Aug 27, 2026

Research, General Agents

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$400K–$500K / yr

Posted Aug 27, 2026

Research, Coding Agents

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$400K–$500K / yr

Posted Aug 27, 2026

Research, Post-Training Evals

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Aug 27, 2026

Research Scientist - Salesforce AI Research

SalesforceCalifornia - Palo Alto +1

Research

Python
PyTorch
JAX
Salesforce

$117K–$345K / yr

Posted Aug 26, 2026

checked 2 days ago

Machine Learning Engineer, Human Centered AI - Evaluations & Insights

AppleSeattle, Washington

Onsite
ML / AI engineering
5+ years

Python
PyTorch
Hugging Face
JAX
MLflow

Pay not posted

Posted Aug 25, 2026

checked 9 days ago

Senior AI Engineer (Risk & Payments)

AirwallexSG - Singapore

Senior
Onsite
ML / AI engineering
2+ years

Python
PyTorch
JAX
LangChain

Pay not posted

Posted Aug 25, 2026

Research, Safety

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Aug 24, 2026

Deep Learning Research Intern - 2027 - Sydney

IMCSydney, Australia +3

Intern
Research

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Aug 24, 2026

S

Distributed Training and Inference Engineer

SciforiumSan Francisco, CA

Onsite
Inference & ML systems
5+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$190K–$250K / yr

Posted Aug 24, 2026

Head of Toronto AI Center

Samsung Research America101 College Street, Toronto, Canada +1

Director+
Software engineering
Manager
10+ years

Python
Java
C++
PyTorch
C
+2 more

Pay not posted

Posted Aug 24, 2026

Founding Member of Technical Staff

Onsite
Inference & ML systems
Member of Technical Staff
Visa sponsorship in the job text

Python
PyTorch
JAX

$136K–$395K / yr

Posted Aug 24, 2026

Campus AI/ML Researcher (Intern)

Jump TradingHong Kong +2

Intern
Research

Python
C++
PyTorch
JAX
TensorFlow

Pay not posted

Posted Aug 24, 2026

Research, RL Scaling

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow
vLLM

$350K–$475K / yr

Posted Aug 21, 2026

TL

Research Engineer

Tessera LabsSan Jose Office (HQ) +1

Research

Python
PyTorch
JAX

$200K–$300K / yr

Posted Aug 21, 2026

G

Founding Research Scientist, Robot Learning

GRAMSan Francisco Bay Area, California +1

Research

Python
C++
PyTorch
JAX

$225K–$300K / yr

Posted Aug 21, 2026

Senior/Staff Applied Scientist, Numerical Optimization & Quantization

NeurophosAustin, Texas +1

Staff
Onsite
Research
5+ years

PyTorch
JAX
TensorFlow

$170K–$240K / yr

Posted Aug 21, 2026

Go to the first page Show the next 40 jobs