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

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

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

Systems Engineer
General RoboticsRedmond, WA
- Infra, platform & SRE
- 1+ years
- Python
- AWS
- Kubernetes
- GCP
- Azure
- +8 more
$155K–$205K / yr
Posted Feb 25, 2026
Research Scientist: Hierarchical Sensorimotor Perception
Astera InstituteEmeryville HQ
- Hybrid
- Research
- PyTorch
- JAX
$150K–$300K / yr
Posted Feb 25, 2026
Research Scientist: Hierarchical RL Agents
Astera InstituteEmeryville HQ
- Hybrid
- Research
- PyTorch
- JAX
$150K–$300K

/ yr

Posted Feb 25, 2026

Research Scientist/Engineer

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

Research Scientist: Energy Based Models

Astera InstituteEmeryville HQ

Hybrid
Research

PyTorch
JAX

$150K–$300K / yr

Posted Feb 25, 2026

EI

AI Compiler and Performance Engineer

ElastixAI INC.Seattle

Hybrid
Inference & ML systems

Python
C++
PyTorch
FPGA
JAX
+1 more

Pay not posted

Posted Feb 24, 2026

Staff Data Scientist | NLP

MachinifyRemote - US

Staff
Remote
ML / AI engineering
3+ years

Python
AWS
SQL
GCP
Azure
+4 more

$180K–$230K / yr

Posted Feb 24, 2026

CH

ML Research Engineer

Circadia HealthVauxhall, London

Onsite
Research

Python
AWS
GCP
Azure
Docker
+5 more

Pay not posted

Posted Feb 23, 2026

Machine Learning Engineer - IV (Biometrics)

JumioIndia (remote) +1

Remote
ML / AI engineering
5+ years

Python
AWS
PyTorch
Airflow
JAX
+1 more

Pay not posted

Posted Feb 23, 2026

CP

AI Solutions Architect | US

Cuesta PartnersRemote US

Remote
Solutions engineering
8+ years

AWS
Kubernetes
GCP
Azure
Docker
+12 more

$145K–$188K / yr

Posted Feb 22, 2026

EI

Machine Learning Research Scientist

ElastixAI INC.Seattle

Hybrid
Research
3+ years

Python
AWS
Kubernetes
GCP
Azure
+5 more

Pay not posted

Posted Feb 19, 2026

EI

Machine Learning Engineer

ElastixAI INC.Seattle

Hybrid
ML / AI engineering
3+ years

Python
AWS
Kubernetes
GCP
Azure
+5 more

Pay not posted

Posted Feb 19, 2026

Deep Learning Researcher

IMCHong Kong, Hong Kong +3

Research
3+ years

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Feb 19, 2026

Senior Scientist, Machine Learning

Altos LabsSan Francisco Bay Area, CA +2

Senior
ML / AI engineering
5+ years

Python
PyTorch
Hugging Face
JAX

$179K–$330K / yr

Posted Feb 17, 2026

LL

Runtime Engineer

Lemurian LabsSanta Clara, CA - Toronto, Canada +3

Remote
Software engineering
4+ years

C++
PyTorch
C
CUDA
JAX

Pay not posted

Posted Feb 13, 2026

LL

Compiler Optimization Engineer

Lemurian LabsSanta Clara, CA - Toronto, Canada +3

Remote
Inference & ML systems
4+ years

C++
PyTorch
C
JAX

Pay not posted

Posted Feb 13, 2026

LL

Front End Compiler

Lemurian LabsSanta Clara, CA - Toronto, Canada +3

Remote
Inference & ML systems
2+ years

Python
C++
PyTorch
C
JAX
+1 more

Pay not posted

Posted Feb 13, 2026

Research Engineer, Machine Learning (Reinforcement Learning)

AnthropicLondon, UK

Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

£260K–£630K / yr

Posted Feb 11, 2026

Machine Learning Performance Engineer

Optiver New York, New York +1

ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+2 more

$200K / yr

Posted Feb 5, 2026

Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

Annapurna Labs (U.S.) Inc.Seattle, Washington +1

Inference & ML systems

AWS
PyTorch
JAX

$144K–$224K / yr

Posted Feb 5, 2026

checked 2 days ago

ET

Principal Solutions Architect (Req #1048)

ePlus Technology, inc.San Ramon, CA +1

Principal
Solutions engineering
8+ years

AWS
Kubernetes
GCP
Azure
PyTorch
+8 more

$170K–$190K / yr

Posted Feb 4, 2026

Software Development Manager - Compiler

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

Inference & ML systems
Manager

AWS
PyTorch
JAX

$213K–$288K / yr

Posted Feb 4, 2026

checked 2 days ago

Quantitative Developer (Python) - Central Liquidity Strategies

MillenniumNew York, New York

Onsite
Quant and trading
5+ years

Python
C++
Rust
PyTorch
CUDA
+2 more

$160K–$250K / yr

Posted Feb 3, 2026

checked 2 days ago

BR

Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics

Bedrock Robotics IncSan Francisco, CA

Hybrid
ML / AI engineering
3+ years

PyTorch
JAX
TensorFlow

Pay not posted

Posted Jan 31, 2026

MR

Research Scientist, Robot Foundation Models

Mind RoboticsPalo Alto

Onsite
Research

Python
PyTorch
JAX

Pay not posted

Posted Jan 26, 2026

MR

Research Engineer, Robot Foundation Models

Mind RoboticsPalo Alto

Onsite
Research

Python
PyTorch
JAX

Pay not posted

Posted Jan 26, 2026

Post-Training Research Scientist (LLMs) — Experimental Track

VettoFlexible (EU-friendly time zones preferred) +1

Research

Python
PyTorch
JAX

Pay not posted

Posted Jan 23, 2026

AI Researcher – Multilingual Data

Featherless AIRemote (world)

Research

Python
PyTorch
JAX

Pay not posted

Posted Jan 23, 2026

AI Researcher — AI Architecture Research

Featherless AIRemote (world)

Research

PyTorch
JAX

Pay not posted

Posted Jan 23, 2026

Machine Learning Engineer — AI Architecture Research

Featherless AIRemote (world)

Research

PyTorch
JAX

Pay not posted

Posted Jan 22, 2026

Member of Technical Staff - Quantitative Research

UdioNew York City (Remote possible for exceptional candidates) +1

Remote
Quant and trading
Member of Technical Staff
5+ years

Kubernetes
GCP
JAX
TensorFlow

$250K–$350K / yr

Posted Jan 22, 2026

Machine Learning Engineer — Distillation

Featherless AIRemote (world)

ML / AI engineering

PyTorch
JAX

Pay not posted

Posted Jan 22, 2026

Machine Learning Engineer (Foundation Models & Personalization)

Eight SleepSan Francisco

Onsite
ML / AI engineering
2+ years

Python
SQL
PyTorch
JAX
Ray
+1 more

Pay not posted

Posted Jan 21, 2026

Staff AI Engineer

JazzX AISan Francisco Bay Area +1

Staff
ML / AI engineering
10+ years

Python
Kubernetes
Java
C++
PyTorch
+3 more

Pay not posted

Posted Jan 20, 2026

D

Reinforcement Learning Engineer

DexmateFremont Office

Onsite
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$120K–$300K / yr

Posted Jan 19, 2026

Head of Engineering, Platform

GoodfireSan Francisco, CA +1

Director+
Infra, platform & SRE
Manager
10+ years

PyTorch
JAX

$300K–$450K / yr

Posted Jan 15, 2026

Internship - Search Machine Learning Engineer

PerplexityBelgrade +1

Intern
ML / AI engineering
2 positions

Python
Rust
PyTorch
A/B testing
JAX
+1 more

Pay not posted

Posted Jan 13, 2026

Member of Technical Staff, Research Engineer

RunwayRemote

Research
Member of Technical Staff
4+ years

PyTorch
JAX

$270K–$370K / yr

Posted Jan 9, 2026

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQUnited States

Staff
Remote
ML / AI engineering
8+ years

Python
AWS
GCP
PyTorch
JAX
+4 more

$157K–$294K / yr

Posted Jan 9, 2026

Staff Product Manager, Training Infrastructure - Weights & Biases

CoreWeaveLivingston, NJ / New York, NY / San Francisco, CA / Sunnyvale, CA / Bellevue, WA +1

Staff
Product management
7+ years

Kubernetes
PyTorch
JAX

$188K–$275K / yr

Posted Jan 8, 2026

Go to the first page Show the next 40 jobs