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 4 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 Engineer, Perception, Simulation Realism
WaymoMountain View, California +2
- Staff
- ML / AI engineering
- Python
- C++
- PyTorch
- JAX
- NumPy
$298K–$368K / yr
Posted 10 days ago
Senior Software Engineer, Machine Learning, Discover Interest Modeling and Retrieval
GoogleMountain View, California
- Senior
- Onsite
- ML / AI engineering
- 5+ years
- Python
- SQL
- C++
- PyTorch
- Spark
- +2 more
$174K–$252K / yr
Posted 10 days ago
checked 8 days ago
Lead ML Engineer

Ooak DataParis, Île-de-France, FR

Senior
ML / AI engineering
8+ years

Python
PyTorch
JAX

€90K–€120K / yr

Posted 11 days ago

checked 2 days ago

Senior Staff Software Engineer, AI/ML, Google Cloud Compute Infrastructure

GoogleSunnyvale, California

Staff
Onsite
ML / AI engineering
8+ years

GCP
PyTorch
JAX
TensorFlow

$262K–$364K / yr

Posted 11 days ago

checked 8 days ago

Robotics, Hardware Engineer, Simulation Engineer

Lumen LabsSan Francisco, CA +1

Onsite
Hardware and robotics

Python
JAX
ROS

$150k–200k + equity

Posted 11 days ago

checked 2 days ago

SS

Staff Engineer, Compiler

Samsung SemiconductorSan Jose, California +1

Staff
Inference & ML systems
10+ years

PyTorch
JAX
TensorFlow

$163K–$253K / yr

Posted 11 days ago

R

Machine Learning Engineer Intern, AI & Robotics

RoboForceMilpitas, CA

Intern
ML / AI engineering

PyTorch
JAX
TensorFlow

$5K–$7K / mo

Posted 11 days ago

Quantum Characterization, Verification, and Validation Engineer

PsiQuantumMilpitas, California

Software engineering

Python
PyTorch
CUDA
JAX
NumPy
+1 more

$89K–$149K / yr

Posted 11 days ago

Machine Learning Engineer - New Grad

SambaNovaAustin, Texas +3

Entry level
ML / AI engineering

Python
PyTorch
JAX
TensorFlow
vLLM

$131K–$197K / yr

Posted 11 days ago

Staff Software Engineer, Server and Full Stack, Growth and Discovery, DeepMind

GoogleNew York, New York

Staff
Onsite
Software engineering
8+ years

Python
Java
Go
C++
PyTorch
+2 more

$207K–$300K / yr

Posted 11 days ago

checked 8 days ago

Software Engineering Manager, GenAI, Computer Vision, Google Maps Immersive Navigation

GoogleMountain View, California

Onsite
ML / AI engineering
Manager
8+ years

Python
Java
C++
JavaScript
C
+2 more

$207K–$300K / yr

Posted 12 days ago

checked 8 days ago

Senior Applied Research Scientist – AI Native Numerical Methods

NVIDIAUS, CA, Remote +4

Senior
Research
5+ years

Python
PyTorch
JAX

$192K–$357K / yr

Posted 12 days ago

checked 5 days ago

Senior Software Development Engineer, AWS Kubernetes (K8s)

Amazon Development Center U.S., Inc.Santa Clara, California

Senior
Infra, platform & SRE

AWS
Kubernetes
PyTorch
JAX

$193K–$262K / yr

Posted 12 days ago

checked 2 days ago

Machine Learning Engineer, Bellwether, X

XMountain View, CA (HQ)

ML / AI engineering
5+ years

Python
Kubernetes
Docker
Terraform
PyTorch
+4 more

$174K–$261K / yr

Posted 12 days ago

Staff Machine Learning Engineer, Sensors & Intelligence

TonalToronto, ON

Staff
Remote
ML / AI engineering

PyTorch
JAX
TensorFlow

CA$160K–CA$185K / yr

Posted 12 days ago

Research Engineer, Infrastructure

HarveyNew York +3

Hybrid
Research
4 positions

Python
PyTorch
JAX

$230K–$450K / yr

Posted 12 days ago

Sr. Model Engineer, Post-Training & RL

xAIPalo Alto, CA

Senior
Inference & ML systems

Python
C++
Rust
PyTorch
JAX

$220K–$357K / yr

Posted 12 days ago

Sr. Model Engineer, Mid-Training

xAIPalo Alto, CA

Senior
Software engineering

Python
Docker
C++
Rust
PyTorch
+3 more

$220K–$357K / yr

Posted 12 days ago

Senior Software Engineer, AI/ML, Geo and Gemini App

GoogleNew York, New York

Senior
Onsite
ML / AI engineering
5+ years

C++
PyTorch
A/B testing
JAX
TensorFlow

$174K–$252K / yr

Posted 13 days ago

checked 8 days ago

Senior Deep Learning Frameworks CUDA Software Engineer

NVIDIAUS, CA, Santa Clara +2

Senior
Inference & ML systems
8+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$184K–$357K / yr

Posted 13 days ago

checked 5 days ago

Senior Compiler Engineer - AI

NVIDIAUS, CA, Santa Clara +4

Senior
Inference & ML systems
6+ years
2 positions

Python
C++
PyTorch
C
JAX

$184K–$288K / yr

Posted 13 days ago

checked 5 days ago

Senior Deep Learning Compiler Engineer - PyTorch

NVIDIAGermany, Berlin +2

Senior
Inference & ML systems
8+ years

Python
PyTorch
JAX

€101K–€194K / yr

Posted 15 days ago

checked 5 days ago

Deep Learning Compiler Verification Engineer

NVIDIAUS, CA, Santa Clara +3

Inference & ML systems
2+ years

Python
C++
PyTorch
JAX

$116K–$190K / yr

Posted 15 days ago

checked 5 days ago

Senior Scientist, Machine Learning Research

A-Alpha BioSeattle, Washington +1

Senior
Research
12+ years

AWS
PyTorch
JAX

$164K–$206K / yr

Posted 15 days ago

Staff Research Scientist, Foundational Audio

GoogleMountain View, California +1

Staff
Onsite
Research
2+ years

PyTorch
Hugging Face
JAX
TensorFlow

$174K–$252K / yr

Posted 15 days ago

checked 8 days ago

Senior Staff AI Engineer

JazzX AIBangalore +1

Staff
ML / AI engineering
10+ years

Python
Kubernetes
Java
C++
PyTorch
+3 more

$207K–$290K / yr

Posted 16 days ago

GI

Member of Technical Staff – Infrastructure

General Intuition & MedalNew York City

Onsite
Inference & ML systems
Member of Technical Staff
6+ years

Kubernetes
GCP
C++
Rust
Terraform
+2 more

$300K–$500K / yr

Posted 16 days ago

Solutions Architect, LLM Model Builder

NVIDIAUS, CA, Santa Clara

Solutions engineering
5+ years

Python
PyTorch
CUDA
JAX
TensorFlow
+1 more

$152K–$242K / yr

Posted 16 days ago

checked 5 days ago

Senior Solutions Architect, Robotics Foundation Model Training

NVIDIAUS, CA, Santa Clara +1

Senior
Solutions engineering
5+ years

PyTorch
Hugging Face
JAX

$152K–$288K / yr

Posted 16 days ago

checked 5 days ago

Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIAUS, CA, Santa Clara

Senior
Research
10+ years

Python
Kubernetes
C++
PyTorch
CUDA
+2 more

$224K–$357K / yr

Posted 16 days ago

checked 5 days ago

Senior Robotics Research Scientist

NVIDIAUS, WA, Seattle

Senior
Research
3+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$192K–$357K / yr

Posted 16 days ago

checked 5 days ago

Senior Software Engineer, JAX

NVIDIAUS, CA, Santa Clara +1

Senior
Software engineering
6+ years

Python
C++
PyTorch
C
JAX
+2 more

$184K–$357K / yr

Posted 16 days ago

checked 5 days ago

Senior Research Engineer - Autonomous Vehicles

NVIDIAUS, CA, Santa Clara

Senior
Research
10+ years

Python
Kubernetes
C++
PyTorch
CUDA
+2 more

$184K–$357K / yr

Posted 16 days ago

checked 5 days ago

Research Scientist, Robotics Research - PhD New College Grad 2026

NVIDIAUS, WA, Seattle

Entry level
Research

Python
C++
PyTorch
CUDA
JAX
+1 more

$168K–$265K / yr

Posted 16 days ago

checked 5 days ago

Director, Perception - Autonomous Vehicles

NVIDIAUS, CA, Santa Clara

Director+
ML / AI engineering
Manager
5+ years

Python
C++
PyTorch
JAX
TensorFlow

$320K–$489K / yr

Posted 16 days ago

checked 5 days ago

Senior Manager, Software Engineering - JAX

NVIDIAUS, CA, Santa Clara

Senior
Software engineering
Manager
4+ years

CUDA
JAX

$272K–$431K / yr

Posted 16 days ago

checked 5 days ago

Senior Robotics Research Engineer, Robotics and AI for Drug Discovery

NVIDIAUS, CA, Santa Clara

Senior
Research
3+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$184K–$357K / yr

Posted 16 days ago

checked 5 days ago

Principal AI and ML Infra Software Engineer, GPU Clusters

NVIDIAUS, CA, Santa Clara +1

Principal
Inference & ML systems
15+ years

Python
AWS
Kubernetes
GCP
Azure
+5 more

$272K–$431K / yr

Posted 16 days ago

checked 5 days ago

Senior Software Engineer — cuEquivariance

NVIDIAUS, CA, Santa Clara

Senior
Software engineering
6+ years

Python
C++
PyTorch
CUDA
JAX

$184K–$357K / yr

Posted 16 days ago

checked 5 days ago

Senior Software Engineer, Spatial Intelligence and Foundation Models

NVIDIAUS, CA, Santa Clara

Senior
Software engineering
8+ years

Python
C++
PyTorch
JAX
TensorFlow

$184K–$357K / yr

Posted 16 days ago

checked 5 days ago

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