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.

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

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

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

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

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

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

Partner Engineer, Generative AI

MetaMenlo Park, California

Onsite
Software engineering
2+ years

Python
AWS
GCP
Azure
Java
+7 more

$149K–$210K / yr

Posted Aug 21, 2026

checked 9 days ago

AI Engineer (interested in physical infrastructure)

TorusSan Francisco, California

ML / AI engineering

Python
TypeScript
C++
Rust
PyTorch
+1 more

$150K–$250K / yr

Posted Aug 20, 2026

checked 2 days ago

F

Data Scientist

FuseSan Leandro, California

Data science

Python
C++
PyTorch
JAX
NumPy

$145K–$190K / yr

Posted Aug 19, 2026

Forward Deployed Engineer (Training)

BasetenSan Francisco +1

Hybrid
Software engineering
1+ years

Kubernetes
PyTorch
JAX
vLLM

$200K–$400K / yr

Posted Aug 19, 2026

C

Research Scientist

ConstellationSan Francisco

Onsite
Research

Python
PyTorch
JAX

$180K–$300K / yr

Posted Aug 18, 2026

Senior/Staff Software Engineer, Infrastructure (ML)

Nimble RoboticsSan Francisco, CA +1

Staff
ML / AI engineering
4+ years

Python
Kubernetes
Go
C++
Rust
+3 more

$210K–$300K / yr

Posted Aug 18, 2026

DM

Senior Staff LLM Inference Engineer

d-MatrixSanta Clara

Staff
Hybrid
Inference & ML systems
10+ years

Python
C++
C
CUDA
JAX
+1 more

$195K–$285K / yr

Posted Aug 18, 2026

Applied Scientist -- Foundation Models, SSO

Amazon Web Services, Inc.Santa Clara, California

Research

AWS
PyTorch
CUDA
JAX

$172K–$222K / yr

Posted Aug 17, 2026

checked 2 days ago

Applied Scientist -- Foundation Models, SSO

Amazon Web Services, Inc.Houston, Texas

Research

AWS
PyTorch
CUDA
JAX

$143K–$193K / yr

Posted Aug 17, 2026

checked 2 days ago

Research Scientist, Theory & Computation

Astera InstituteAlameda

Onsite
Research
3+ years

Python
PyTorch
JAX

$150K–$300K / yr

Posted Aug 16, 2026

Principal Applied Scientist - AI

UiPathBellevue

Principal
Onsite
Research

Python
Kubernetes
Azure
PyTorch
CUDA
+3 more

$218K–$265K / yr

Posted Aug 14, 2026

F

Helix AI Engineer, Training Performance

FigureSan Jose, CA +1

ML / AI engineering
3+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$200K–$400K / yr

Posted Aug 14, 2026

Perception Machine Learning Engineer - Continuous Learning

WaymoMountain View, CA +2

ML / AI engineering
3+ years

Python
PyTorch
JAX

$175K–$215K / yr

Posted Aug 13, 2026

Member of Technical Staff, Agentic Environments

CohereEurope +2

Remote
Inference & ML systems
Member of Technical Staff

Python
Kubernetes
PyTorch
JAX

CA$250K–CA$535K / yr

Posted Aug 13, 2026

I

Staff GNC Engineer (State Estimation)

InversionPlaya Vista, California +1

Staff
Software engineering
9+ years

Python
C++
PyTorch
JAX
MATLAB

$161K–$221K / yr

Posted Aug 13, 2026

Compiler Engineer, Graph Compiler Performance Optimization

MetaMenlo Park, California

Onsite
Inference & ML systems

PyTorch
JAX
TensorFlow

$184K–$257K / yr

Posted Aug 12, 2026

checked 9 days ago

Machine Learning Performance Engineer (Inference)

Tower Research CapitalNew York

Inference & ML systems
2+ years

Python
C++
PyTorch
FPGA
JAX

$200K–$300K / yr

Posted Aug 11, 2026

A

AI Engineer

ArmadaBellevue Office, Sunset Corporate Campus

ML / AI engineering
3+ years

Python
Kubernetes
Java
C++
PyTorch
+4 more

$155K–$193K / yr

Posted Aug 10, 2026

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

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

Senior
Inference & ML systems

AWS
PyTorch
JAX

$168K–$262K / yr

Posted Aug 10, 2026

checked 2 days ago

Staff/Lead Machine Learning Research Scientist, Behavior & Planning

NuroMountain View, California (HQ) +1

Staff
Research
7+ years

Python
C++
PyTorch
JAX
TensorFlow

$235K–$352K / yr

Posted Aug 7, 2026

Computational Scientist, Electric Grid Simulations, Tapestry

XMountain View, CA (HQ)

Data science
2+ years

Python
Java
C++
C
JAX
+1 more

$147K–$221K / yr

Posted Aug 7, 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