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,867 companies, checked on their job boards 29 min 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.

AI Systems, Language & Reasoning Models
Unconventional AIMountain View, CA I US Remote +2
- Remote
- ML / AI engineering
- PyTorch
- JAX
Pay not posted
Posted Apr 14, 2026
Research Engineer, Model Inference & Serving - Paris
H CompanyHybrid Paris
- Hybrid
- Research
- Python
- Kubernetes
- Go
- C++
- Rust
- +4 more
Pay not posted
Posted Apr 14, 2026
Research Engineer, Bellwether, X
XMountain View, CA (HQ)
- Research
- 5+ years
- Python
- PyTorch
- JAX
- NumPy
- pandas
- +1 more

$174K–$261K / yr

Posted Apr 13, 2026

Member of Technical Staff (AI Inference Engineer)

PerplexitySan Francisco +2

Inference & ML systems
Member of Technical Staff
3+ years

Python
Rust
PyTorch
CUDA
JAX
+1 more

$220K–$485K / yr

Posted Apr 13, 2026

Member of Technical Staff (AI Inference Engineer)

PerplexityLondon

Inference & ML systems
Member of Technical Staff
3+ years

Python
Kubernetes
Rust
PyTorch
CUDA
+2 more

Pay not posted

Posted Apr 13, 2026

AI/ML Research Engineer

Manifold BioBoston, MA or San Francisco, CA +2

Research
2+ years

Python
AWS
Kubernetes
GCP
Docker
+5 more

$140K–$225K / yr

Posted Apr 13, 2026

Senior Research Machine Learning Engineer

PhysicsXLondon, United Kingdom +1

Senior
Research

Python
AWS
GCP
Azure
C++
+5 more

Pay not posted

Posted Apr 13, 2026

Research Engineer, Model Inference & Serving - London

H CompanyHybrid London

Hybrid
Research

Python
Kubernetes
Go
C++
Rust
+4 more

Pay not posted

Posted Apr 10, 2026

N

Member of Technical Staff [Research]

NeoCognitionPalo Alto

Onsite
Research
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Apr 9, 2026

Member of Technical Staff

Axiom MathPalo Alto +4

Onsite
ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Apr 3, 2026

Research Engineer, World Models

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX
Ray

$155K–$269K / yr

Posted Apr 2, 2026

Staff Software Engineer - Backend

WalmartIN KA BANGALORE Home Office PTPP2

Staff
Software engineering
12+ years

SQL
React
Java
JavaScript
Elasticsearch
+6 more

Pay not posted

Posted Apr 2, 2026

checked 2 days ago

System Modeling (Dynamic Systems Simulation)

Unconventional AIMountain View, CA I US Remote +2

Remote
Infra, platform & SRE

Python
PyTorch
CUDA
JAX
NumPy
+1 more

Pay not posted

Posted Apr 1, 2026

XT

BioMedical AI Research Engineer

Xaira TherapeuticsSouth San Francisco, California +1

Research

Python
PyTorch
JAX
TensorFlow

$252K–$308K / yr

Posted Mar 31, 2026

Systems Engineer - Machine Learning

General RoboticsRedmond, WA

Onsite
ML / AI engineering
1+ years

Python
AWS
Kubernetes
GCP
Azure
+3 more

$155K–$200K / yr

Posted Mar 30, 2026

Machine Learning Researcher - Springtail

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
CUDA
JAX

$150K–$300K / yr

Posted Mar 29, 2026

Research Scientist, World Models

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX

$155K–$269K / yr

Posted Mar 27, 2026

Research Scientist, Neural Reconstruction

WaabiToronto, ON +3

Hybrid
Research

Python
PyTorch
JAX

$155K–$269K / yr

Posted Mar 27, 2026

Senior / Staff Software Engineer, ML Datasets & Data Pipelines

WaabiToronto, ON +2

Staff
Hybrid
ML / AI engineering
4+ years

Python
AWS
GCP
PyTorch
Spark
+4 more

$148K–$260K / yr

Posted Mar 26, 2026

DU

Senior/Staff Deep Reinforcement Learning Engineer - DoorDash Dot

DoorDash USASan Francisco, CA

Staff
ML / AI engineering

JAX

$168K–$247K / yr

Posted Mar 25, 2026

Senior Applied ML Engineer

CoreWeaveBellevue, WA / Sunnyvale, CA +2

Senior
ML / AI engineering
4+ years

Python
Kubernetes
PyTorch
PostgreSQL
FastAPI
+1 more

$182K–$242K / yr

Posted Mar 25, 2026

VF

Machine Learning Researcher

Virtu FinancialNew York

Research
2+ years

Python
PyTorch
JAX

$200K–$300K / yr

Posted Mar 24, 2026

Senior Member of Technical Staff, Safety and Security for Agents

CohereLondon +5

Senior
Hybrid
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

CA$250K–CA$535K / yr

Posted Mar 23, 2026

LA

Senior Data Scientist (US)

Lynx AnalyticsNew York +4

Senior
Data science
8+ years

Python
AWS
SQL
GCP
Azure
+4 more

Pay not posted

Posted Mar 19, 2026

A

AMI Research Engineer

Onsite
Research

Python
PyTorch
JAX

Pay not posted

Posted Mar 19, 2026

A

AMI Scientist

Onsite
Data science

Python
PyTorch
JAX

Pay not posted

Posted Mar 19, 2026

B

AI Researcher

BeyondMathLondon

Hybrid
Research
5+ years

PyTorch
JAX
TensorFlow

Pay not posted

Posted Mar 18, 2026

Machine Learning Engineer, Senior

9 MothersAustin, Texas

Senior
ML / AI engineering

Python
C++
Rust
PyTorch
JAX

$150K–$250K / yr

Posted Mar 18, 2026

checked 2 days ago

Research Engineer

Nace AIPalo Alto, CA

Onsite
Research
3+ years

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Mar 17, 2026

Member of Technical Staff, AI Bio

Radical NumericsSan Francisco

Onsite
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX

Pay not posted

Posted Mar 17, 2026

Model Engineer, Voice

xAIPalo Alto, CA

Software engineering

Python
Kubernetes
C++
Rust
PyTorch
+2 more

$160K–$250K / yr

Posted Mar 16, 2026

Tech Lead Manager ML Optimization

WaymoMountain View, California +1

Senior
ML / AI engineering
Manager
10+ years

PyTorch
JAX
TensorFlow

$298K–$378K / yr

Posted Mar 13, 2026

B

Machine Learning Engineer

BeyondMathLondon

Hybrid
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Mar 13, 2026

Machine Learning Engineer

Tower Research CapitalHong Kong, Singapore, Shanghai +3

ML / AI engineering
2+ years

Python
Kubernetes
Docker
PyTorch
JAX
+1 more

Pay not posted

Posted Mar 13, 2026

O

Member of Technical Staff, ML Performance

OdysseyPalo Alto +2

Hybrid
Inference & ML systems
Member of Technical Staff
8+ years

PyTorch
JAX

$149K–$462K / yr

Posted Mar 11, 2026

Research Systems Engineer

Applied ComputeSan Francisco

Onsite
Research
Visa sponsorship in the job text

PyTorch
JAX

$190K–$310K / yr

Posted Mar 10, 2026

Machine Learning Engineer

Hestus, Inc.San Mateo, California

ML / AI engineering
4+ years

Python
AWS
SQL
GCP
Azure
+5 more

$180K–$240K / yr

Posted Mar 9, 2026

checked 2 days ago

M

Expression of Interest: Machine Learning Engineer

MolocoMenlo Park, California +5

ML / AI engineering
5+ years

Python
AWS
Kubernetes
GCP
Go
+6 more

$170K–$386K / yr

Posted Mar 6, 2026

HA

Data Engineer - Foundational

Harmattan AIParis

Onsite
Data engineering
5+ years

PyTorch
Spark
JAX
Ray

Pay not posted

Posted Mar 6, 2026

P

Research Engineer

PhonicSan Francisco

Onsite
Research

Python
PyTorch
JAX

Pay not posted

Posted Mar 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