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

RA
Software Engineer, River API
River AI Inc.Palo Alto, CA +1
- Software engineering
- Visa sponsorship in the job text
- C++
- Rust
- PyTorch
- C
- JAX
$200K–$420K / yr
Posted May 14, 2026
Research Scientist - Frontier AI/ML & Quantum Algorithms
Sygaldry TechnologiesSan Francisco
- Onsite
- Research
- Visa sponsorship in the job text
- Python
- PyTorch
- JAX
$200K–$300K / yr
Posted May 13, 2026
Machine Learning Intern/Co-op (Winter 2027)
CohereCanada +3

Intern
Remote
ML / AI engineering

Python
CUDA
JAX
TensorFlow

Pay not posted

Posted May 13, 2026

Perception Engineer, Staff

9 MothersAustin, Texas

Staff
ML / AI engineering
6+ years

Python
C++
Rust
PyTorch
JAX

$250K–$300K / yr

Posted May 13, 2026

checked 2 days ago

RA

Software Engineer, Inference Infrastructure

Rhoda AIMountain View

Inference & ML systems
3+ years

AWS
Kubernetes
GCP
PyTorch
gRPC
+3 more

$175K–$280K / yr

Posted May 12, 2026

Software Engineer, Perception

IntrinsicSingapore +1

ML / AI engineering
1+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted May 11, 2026

Machine Learning Engineer, Speech LLM Training - San Francisco

PlaudSan Francisco, CA +1

Hybrid
ML / AI engineering

PyTorch
JAX

$195K–$365K / yr

Posted May 8, 2026

H

Research Engineer, Training & Inference

HarmonicPalo Alto

Onsite
Research
2+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$200K–$450K / yr

Posted May 7, 2026

CS

Artificial Intelligence/Machine Learning SME

Careers SciresAugusta, GA

ML / AI engineering
12+ years

Python
Kubernetes
GCP
Azure
Docker
+11 more

Pay not posted

Posted May 7, 2026

checked 2 days ago

Senior / Staff Machine Learning Research Engineer

Calico South San Francisco, CA

Staff
Research
7+ years

Python
Kubernetes
GCP
Docker
PyTorch
+1 more

$220K–$290K / yr

Posted May 5, 2026

PM

Member of Technical Staff - Machine Learning Capabilities, New Graduates

Preference ModelSan Francisco

Entry level
Onsite
ML / AI engineering
Member of Technical Staff
Visa sponsorship in the job text

Python
AWS
GCP
Azure
PyTorch
+3 more

$165K–$200K / yr

Posted May 4, 2026

BR

Machine Learning Engineer: Perception

Bedrock Robotics IncSan Francisco, CA

Hybrid
ML / AI engineering
5+ years

Python
C++
Rust
PyTorch
JAX
+1 more

Pay not posted

Posted May 4, 2026

Research Internship (Winter 2027)

CohereCanada +7

Intern
Remote
Research

Python
C++
PyTorch
C
JAX
+1 more

Pay not posted

Posted May 1, 2026

N

Research Intern

NeoCognitionPalo Alto

Intern
Hybrid
Research

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted May 1, 2026

Performance Engineer (Inference, Training & GPU)

World LabsSan Francisco

Onsite
Inference & ML systems
Visa sponsorship in the job text

Python
Go
C++
Rust
PyTorch
+2 more

$200K–$300K / yr

Posted May 1, 2026

A

Machine Learning Research Engineer (MLRE) - GPUs

AchiraSan Francisco Office +2

Hybrid
Research
2+ years

PyTorch
CUDA
JAX

$165K–$259K / yr

Posted Apr 29, 2026

A

Machine Learning Research Engineer (MLRE) - Workflows/Systems

AchiraSan Francisco Office +1

Hybrid
Research
2+ years

PyTorch
JAX

$165K–$259K / yr

Posted Apr 29, 2026

A

Machine Learning Research Engineer (MLRE) - Research

AchiraSan Francisco Office +1

Hybrid
Research

PyTorch
JAX

$165K–$259K / yr

Posted Apr 29, 2026

Senior Staff Machine Learning Engineer - Content Platform

SpotifyNew York, NY

Staff
Remote
ML / AI engineering

PyTorch
JAX
TensorFlow

$281K–$402K / yr

Posted Apr 29, 2026

Senior Machine Learning Engineer, Runtime and Serving

WaymoMountain View, CA +1

Senior
Inference & ML systems
5+ years

Python
C++
PyTorch
CUDA
JAX

$213K–$263K / yr

Posted Apr 29, 2026

Model Engineer, Post-Training & RL

xAIPalo Alto, CA

Inference & ML systems

Python
C++
Rust
PyTorch
JAX

$160K–$250K / yr

Posted Apr 29, 2026

TA

Research Engineer, Robotics: Physics Simulation and Sim-to-Real

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research
3+ years

Python
C++
CUDA
JAX

$129K–$193K / yr

Posted Apr 28, 2026

AI Research Engineer, Post-Training

Onsite
Research

PyTorch
JAX

Pay not posted

Posted Apr 28, 2026

Senior Staff Machine Learning Engineer - Content Platform

SpotifyLondon +1

Staff
Hybrid
ML / AI engineering

PyTorch
JAX
TensorFlow

Pay not posted

Posted Apr 28, 2026

LS

Senior / Principal ML Scientist, Foundation Models for Life Sciences

Lila SciencesSan Francisco, CA USA +1

Principal
ML / AI engineering

AWS
GCP
PyTorch
JAX
TensorFlow

$268K–$384K / yr

Posted Apr 27, 2026

AI

Director, Machine Learning, Virtual Cell Initiative

Arc InstitutePalo Alto, CA +1

Director+
ML / AI engineering
Manager
5+ years

PyTorch
JAX
TensorFlow

$380K–$420K / yr

Posted Apr 27, 2026

Senior Research Scientist

PhysicsXLondon, United Kingdom +1

Senior
Research
4+ years

Python
PyTorch
CUDA
JAX
NumPy

Pay not posted

Posted Apr 27, 2026

9M

Perception Engineer

9 MothersAustin

Onsite
ML / AI engineering
6+ years

Python
C++
Rust
PyTorch
JAX

$110K–$300K / yr

Posted Apr 23, 2026

Research Engineer, Machine Learning (RL Velocity)

AnthropicLondon, UK

Research
Visa sponsorship in the job text

PyTorch
JAX

£370K–£630K / yr

Posted Apr 23, 2026

Research Engineer, Machine Learning (RL Velocity)

AnthropicRemote-Friendly (Travel-Required) +2

Remote
Research
Visa sponsorship in the job text

PyTorch
JAX

$500K–$850K / yr

Posted Apr 23, 2026

E

Research Scientist - Machine Learning

ExtropicBoston +1

Onsite
Research
2 positions

Python
AWS
PyTorch
JAX
TensorFlow

$150K–$250K / yr

Posted Apr 22, 2026

Senior Machine Learning Engineer - (Biometrics)

JumioMontreal

Senior
ML / AI engineering
5+ years

Python
AWS
PyTorch
Airflow
JAX
+1 more

Pay not posted

Posted Apr 22, 2026

F

Research Engineer

ForusNew York office

Research

PyTorch
JAX
TensorFlow

$170K–$300K / yr

Posted Apr 22, 2026

Sr. Software Engineer, Computer Vision

SpaceXHawthorne, CA

Senior
ML / AI engineering
8+ years

Python
Kubernetes
Docker
C++
PyTorch
+11 more

$160K–$225K / yr

Posted Apr 21, 2026

H

Robot Learning Engineer

HumanoidUK, London

Onsite
Software engineering
3+ years

PyTorch
JAX

Pay not posted

Posted Apr 17, 2026

Senior Cluster Site Reliability Engineer

The Voleon GroupBerkeley, CA +1

Senior
Remote
Infra, platform & SRE
5+ years

Python
AWS
GCP
Docker
Terraform
+9 more

$205K–$235K / yr

Posted Apr 17, 2026

Strategic Technical Account Manager GPU

VultrRemote - United States

Remote
Solutions engineering
2+ years

Kubernetes
PyTorch
CUDA
Hugging Face
JAX
+3 more

$115K–$140K / yr

Posted Apr 16, 2026

Z

Principal Architect, AI/ML

ZencoreRemote +1

Principal
ML / AI engineering
2 positions

AWS
GCP
Azure
PyTorch
JAX
+2 more

Pay not posted

Posted Apr 16, 2026

W

Senior Forward Deployed AI Engineer

webAIWashington D.C. Area +1

Senior
Hybrid
ML / AI engineering
5+ years

PyTorch
JAX
TensorFlow

Pay not posted

Posted Apr 15, 2026

Forward Deployed Engineer

GoodfireSan Francisco, CA +1

Software engineering

Python
PyTorch
JAX

$200K–$325K / yr

Posted Apr 14, 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