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.

Research Scientist - Frontier AI/ML & Quantum Algorithms
Sygaldry TechnologiesSan Francisco
- Research
- Visa sponsorship in the job text
- Python
- PyTorch
- CUDA
- JAX
$200K–$300K / yr
Posted Aug 6, 2026
checked 2 days ago
S
GPU Cluster Engineer, Systems & Platform
SciforiumSan Francisco, CA
- Onsite
- Inference & ML systems
- 5+ years
- Python
- Kubernetes
- Docker
- Terraform
- PyTorch
- +8 more
$150K–$220K / yr
Posted Aug 6, 2026
Principal Scientist, Machine Learning

Flagship Pioneering, Inc.Cambridge, MA USA +1

Principal
ML / AI engineering
5+ years

Python
PyTorch
JAX
LangChain
TensorFlow

$208K–$286K / yr

Posted Aug 5, 2026

Research Engineer, Infrastructure, Inference

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Kubernetes
PyTorch
JAX
Ray
vLLM

$400K–$500K / yr

Posted Aug 4, 2026

Research, Pre-Training Data

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Aug 4, 2026

Research, Audio Expertise

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$400K–$500K / yr

Posted Aug 4, 2026

Research Engineer, Infrastructure, Numerics

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

PyTorch
JAX

$400K–$500K / yr

Posted Aug 4, 2026

Research, Vision Expertise

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Aug 4, 2026

Software Engineer, Supercomputing

Thinking Machines LabSan Francisco

Onsite
Inference & ML systems
Visa sponsorship in the job text

Python
Kubernetes
Rust
PyTorch
CUDA
+3 more

$350K–$475K / yr

Posted Aug 4, 2026

Research Engineer, Infrastructure, Kernels

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

PyTorch
CUDA
JAX

$400K–$500K / yr

Posted Aug 4, 2026

Research Engineer, Infrastructure, Training Systems

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

PyTorch
JAX

$400K–$500K / yr

Posted Aug 4, 2026

Senior Machine Learning Engineer, Perception

PlusAISanta Clara, CA

Senior
Hybrid
ML / AI engineering
3+ years

Python
PyTorch
JAX
TensorFlow

$150K–$200K / yr

Posted Aug 4, 2026

Behavior Planning Software Engineer

AuroraMountain View, CA +1

Hybrid
Software engineering
4+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$139K–$223K / yr

Posted Aug 4, 2026

Senior Machine Learning Perception Engineer - Fallback Driving System

General MotorsGM Automation - Sunnyvale - GM Automation - Sunnyvale

Senior
ML / AI engineering
3+ years

Python
C++
PyTorch
JAX
ROS
+1 more

$171K–$261K / yr

Posted Aug 4, 2026

checked 2 days ago

Senior ML Software Engineer, Data Plane

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

Senior
ML / AI engineering
7+ years

C++
PyTorch
C
JAX
Linux
+1 more

$193K–$262K / yr

Posted Aug 4, 2026

checked 2 days ago

Staff Research Scientist, Foundation Models Recipes

WaymoMountain View, CA +5

Staff
Research
3+ years

PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted Aug 3, 2026

Senior Research Scientist, AI Foundations Recipes

WaymoOxford, UK +3

Senior
Research
3+ years

Python
PyTorch
JAX
TensorFlow

£123K–£129K / yr

Posted Jul 31, 2026

S

ML Systems - Member of Technical Staff

SimileSan Francisco

Onsite
Inference & ML systems
Member of Technical Staff

Python
PyTorch
CUDA
JAX
vLLM

$200K–$400K / yr

Posted Jul 30, 2026

HL

Machine Learning Engineer - Advanced AI and Cognitive Systems

HRL LaboratoriesCalabasas, CA

Onsite
ML / AI engineering
3+ years

Python
Kubernetes
Docker
PyTorch
JAX
+1 more

$141K–$176K / yr

Posted Jul 29, 2026

P1

AI Research Scientist

P-1 AISan Francisco Bay Area

Hybrid
Research

Python
PyTorch
JAX

$200K–$385K / yr

Posted Jul 29, 2026

Senior Autonomy Engineer

Brain CorpSan Diego, CA

Senior
ML / AI engineering
5+ years

Python
C++
PyTorch
CUDA
JAX
+2 more

$180K–$224K / yr

Posted Jul 25, 2026

Staff Autonomy Engineer

Brain CorpSan Diego, CA

Staff
ML / AI engineering
7+ years

Python
C++
PyTorch
CUDA
JAX
+2 more

$210K–$262K / yr

Posted Jul 25, 2026

O

Member of Technical Staff, Foundation Models

OdysseyPalo Alto +2

Inference & ML systems
Member of Technical Staff
4+ years

PyTorch
JAX

$164K–$508K / yr

Posted Jul 24, 2026

Staff Machine Learning Engineer - Vision-Language Foundation Models

WaymoMountain View, California +1

Staff
ML / AI engineering
8+ years

Python
PyTorch
JAX

$251K–$310K / yr

Posted Jul 23, 2026

Sr. ML Infrastructure Engineer

xAIPalo Alto, CA

Senior
ML / AI engineering
5+ years

Python
C++
Rust
PyTorch
Ansible
+3 more

$180K–$330K / yr

Posted Jul 22, 2026

DD

System Software Engineer - AI

Delos Data IncPalo Alto

Hybrid
Software engineering
5+ years

Python
C++
PyTorch
CUDA
JAX
+2 more

$140K–$220K / yr

Posted Jul 21, 2026

Machine Learning Engineer, Perception

WingPalo Alto, California

ML / AI engineering
3+ years

Python
C++
PyTorch
JAX
TensorFlow

$190K–$202K / yr

Posted Jul 21, 2026

Staff Machine Learning Engineer

NubankPalo Alto

Staff
Hybrid
ML / AI engineering
5+ years

Python
PyTorch
A/B testing
JAX
TensorFlow

$230K–$345K / yr

Posted Jul 20, 2026

DM

Sr. Staff ML Researcher, LLM Algorithmic Optimization

d-MatrixSanta Clara

Staff
Hybrid
Research
5+ years

Python
PyTorch
JAX
TensorFlow

$175K–$265K / yr

Posted Jul 15, 2026

Senior Software Engineer - GPU Kernel Authoring & Optimization

CoreWeaveSunnyvale, CA / Bellevue, WA +2

Senior
Inference & ML systems
5+ years

Python
Kubernetes
C++
PyTorch
CUDA
+2 more

$182K–$242K / yr

Posted Jul 14, 2026

Model Engineer, RL Training Framework

xAIPalo Alto, CA

ML / AI engineering

Python
C++
Rust
PyTorch
JAX

$160K–$250K / yr

Posted Jul 14, 2026

Senior Machine Learning Scientist I/II, Model-Driven Optimization

Generate BiomedicinesSomerville, MA

Senior
ML / AI engineering

Python
PyTorch
JAX

$192K–$265K / yr

Posted Jul 14, 2026

Research Engineer, Privacy

OpenAISan Francisco

Hybrid
Research

PyTorch
JAX

$500K / yr

Posted Jul 14, 2026

ET

Principal Solutions Architect (Req #1341)

ePlus Technology, inc.Irvine, CA +1

Principal
Solutions engineering
8+ years

AWS
Kubernetes
GCP
Azure
PyTorch
+8 more

$170K–$190K / yr

Posted Jul 13, 2026

Member of Technical Staff - Workers AI Applied Research

CloudflareHybrid +4

Hybrid
Research
Member of Technical Staff

Python
C++
PyTorch
JAX
TensorFlow
+1 more

$185K–$266K / yr

Posted Jul 9, 2026

Sr. Model Engineer, RL Inference

xAIPalo Alto, CA

Senior
Inference & ML systems

Python
C++
Rust
PyTorch
JAX
+1 more

$160K–$250K / yr

Posted Jul 6, 2026

G

Research Scientist – Large Tabular Models (LTMs)

GranicaBay Area Office

Onsite
Research

Python
PyTorch
JAX

$160K–$240K / yr

Posted Jul 2, 2026

LS

Senior / Staff Machine Learning Engineer, Applied AI

Lila SciencesCambridge, MA USA +3

Staff
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$180K–$298K / yr

Posted Jul 1, 2026

A

ML Research Scientist (MLRS) - Representation Learning for Molecular AI

AchiraSan Francisco Office +1

Hybrid
Research

Python
PyTorch
JAX

$165K–$259K / yr

Posted Jun 30, 2026

A

ML Research Scientist (MLRS) - Generative AI

AchiraSan Francisco Office +1

Hybrid
Research

Python
PyTorch
JAX

$165K–$259K / yr

Posted Jun 30, 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