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 (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.

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
C
Agent Runtime & Systems - Member of Technical Staff
CallosumLondon
- Onsite
- Inference & ML systems

Member of Technical Staff

Visa sponsorship in the job text

Python
Go
C++
Rust
PyTorch
+3 more

Pay not posted

Posted Aug 20, 2026

V

Member of Technical Staff - ML Performance

VeedaToronto +3

Hybrid
Inference & ML systems
Member of Technical Staff

Python
C++
PyTorch
CUDA
JAX

Pay not posted

Posted Aug 19, 2026

C

Platform Engineer (Machine Learning)

CygnifySingapore

Onsite
ML / AI engineering

Python
PyTorch
JAX
vLLM

Pay not posted

Posted Aug 19, 2026

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

Genesis-World: Core Simulation Engine Engineer

GenesisSan Carlos +3

Onsite
Software engineering
3 positions

Python
C++
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted Aug 19, 2026

C

Research Scientist

ConstellationSan Francisco

Onsite
Research

Python
PyTorch
JAX

$180K–$300K / yr

Posted Aug 18, 2026

Software Engineer, Search and RAG Applications

AppleSanta Clara, California +1

Onsite
Software engineering

Python
Java
Go
C++
PyTorch
+5 more

Pay not posted

Posted Aug 18, 2026

checked 9 days ago

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

ML Engineer - Creator Studio

AppleCupertino, California +4

Onsite
ML / AI engineering
3+ years

Python
PyTorch
Swift
JAX
TensorFlow

Pay not posted

Posted Aug 17, 2026

checked 9 days ago

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

Fixed Income Quantitative Researcher | Trading Team

Jump TradingLondon

Quant and trading

Python
C++
PyTorch
JAX

Pay not posted

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

Machine Learning Specialist

EncoraMakati, Makati, Philippines +1

ML / AI engineering
3+ years

Python
AWS
SQL
Kubernetes
GCP
+5 more

Pay not posted

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

C

Applied AI/ML Engineer (Agents)

CuspAIAmsterdam, NL +3

Hybrid
ML / AI engineering
4+ years

PyTorch
JAX

Pay not posted

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

2027 Shanghai Machine Learning Engineer Summer Internship

Optiver Shanghai, China +1

Intern
ML / AI engineering

PyTorch
JAX

Pay not posted

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

Senior Deep Learning Research Engineer

PlumeraiLondon +1

Senior
Hybrid
Research
5+ years
2 positions
Visa sponsorship in the job text

Python
Kubernetes
GCP
PyTorch
JAX
+2 more

Pay not posted

Posted Aug 10, 2026

Deep Learning Research Engineer

PlumeraiAmsterdam +1

Hybrid
Research
2 positions
Visa sponsorship in the job text

Python
Kubernetes
GCP
PyTorch
JAX
+2 more

Pay not posted

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

G

AI Research Scientist - Early Career (PHD/MS)

GoalyPalo Alto, CA

Entry level
Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX

Pay not posted

Posted Aug 8, 2026

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

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

OR

Robotics Researcher — Manipulation (Omakase Zen)

Omakase RoboticsUS +1

Research
3+ years

PyTorch
JAX

Pay not posted

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

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