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,835 companies, checked on their job boards 5 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.

(Jr) Machine Learning (ML) Developer
LaminarBoston, MA
- Entry level
- Hybrid
- ML / AI engineering
- Python
- AWS
- GCP
- PyTorch
- Databricks
- +5 more
$89K–$141K / yr
Posted Sep 29, 2026
2027 Summer Intern, PhD, Machine Learning Research, Planning/Prediction
WaymoMountain View, CA +1
- Intern
- Research
- Python
- C++
- PyTorch
- C
- JAX
- +1 more
$85 / hr
Posted Sep 28, 2026
Machine Learning Researcher - Agentic Science
PostEraRemote
- Research

Python

PyTorch

JAX

Pay not posted

Posted Sep 28, 2026

ML Algorithm Mapping and Performance Engineer, Core ML

Cerebras SystemsSunnyvale, CA

Hybrid
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX

Pay not posted

Posted Sep 28, 2026

ML Runtime and Kernel Engineer - Core ML

Cerebras SystemsSunnyvale, CA +1

Hybrid
Inference & ML systems

Python
C++
PyTorch
CUDA
JAX

Pay not posted

Posted Sep 28, 2026

Research Engineer, Machine Learning

Mistral AIParis +3

Research
4+ years

Python
Kubernetes
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted Sep 28, 2026

TA

Research Engineer, Climate Modeling

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research
2+ years

Python
GCP
JAX

$147K–$220K / yr

Posted Sep 25, 2026

2027 Summer Intern, MS/PhD, Perception, Machine Learning

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 25, 2026

TA

Research Scientist, Climate Modeling

The Allen Institute for Artificial IntelligenceSeattle, WA +1

Research

Python
JAX

$167K–$261K / yr

Posted Sep 25, 2026

2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 25, 2026

Senior Director / Vice President, Machine Learning (Generative Chemistry)

Flagship Pioneering, Inc.Cambridge, MA USA +1

Director+
ML / AI engineering
Manager
7+ years

Python
AWS
Docker
Terraform
PyTorch
+7 more

$279K–$413K / yr

Posted Sep 25, 2026

Principal Engineer, Efficient GenAI

Advanced Micro Devices, IncSan Jose, California

Principal
Hybrid
Software engineering

PyTorch
JAX
vLLM

Pay not posted

Posted Sep 25, 2026

Applied Science - Research Engineer (Robotics)

Mistral AISingapore

Research

Python
PyTorch
JAX

Pay not posted

Posted Sep 25, 2026

Research Engineer, ML Infrastructure

Chai DiscoverySan Francisco office

Research

Python
PyTorch
JAX

Pay not posted

Posted Sep 24, 2026

Machine Learning Performance Engineer, Training

Tower Research CapitalNew York, NY +1

ML / AI engineering
3+ years

Python
Kubernetes
C++
PyTorch
CUDA
+1 more

Pay not posted

Posted Sep 24, 2026

2027 Summer Intern, PhD, Research Scientist, Foundation Models

WaymoMountain View, CA +1

Intern
Research

Python
JAX
TensorFlow

$85 / hr

Posted Sep 24, 2026

Staff ML Infrastructure Engineer

AppleCupertino, California

Staff
Onsite
ML / AI engineering
10+ years

Python
Kubernetes
Docker
Go
C++
+6 more

Pay not posted

Posted Sep 24, 2026

checked 9 days ago

SI

Machine Learning Engineering Internship: Summer 2027

Susquehanna International Group, LLPUnited States

Intern
Onsite
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX

Pay not posted

Posted Sep 24, 2026

checked 9 days ago

AIML - Distinguished Engineer, Foundation Model

AppleCupertino, California

Principal
Onsite
Software engineering
15+ years

AWS
Kubernetes
GCP
Azure
Docker
+2 more

Pay not posted

Posted Sep 22, 2026

checked 9 days ago

2027 Summer Intern, MS/PhD, Machine Learning Engineer

WaymoSan Francisco, CA +1

Intern
ML / AI engineering

Python
C++
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Sep 22, 2026

GD

Senior Manager, Applied AI Research

Gen Digital Inc.CZE - Prague +1

Senior
Hybrid
Research
Manager
5+ years

Python
Kubernetes
PyTorch
Hugging Face
JAX
+2 more

Pay not posted

Posted Sep 22, 2026

Staff ML Engineer, VLA Training

Nidus TechnologiesNew York City

Staff
Onsite
ML / AI engineering
5+ years

PyTorch
JAX
ROS

$250K–$350K / yr

Posted Sep 22, 2026

Staff Software Engineer, Machine Learning (Consumer Revenue)

DiscordSan Francisco Bay Area +1

Staff
ML / AI engineering
8+ years

Python
PyTorch
Spark
Flink
JAX
+2 more

$272K–$340K / yr

Posted Sep 22, 2026

Research Scientist, Contextual AI and Multimodal Agents

MetaRedmond, Washington

Onsite
Research
12+ years

Python
PyTorch
JAX

$219K–$301K / yr

Posted Sep 22, 2026

checked 9 days ago

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

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

Senior
Inference & ML systems

AWS
PyTorch
JAX

$193K–$262K / yr

Posted Sep 22, 2026

checked 2 days ago

Sr. Software Engineer- AI/ML, Amazon Neuron Training

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

Senior
ML / AI engineering

PyTorch
JAX

$193K–$262K / yr

Posted Sep 22, 2026

checked 2 days ago

2027 Summer Intern, MS/PhD, Software Engineer, Simulation Evaluation ML Model

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
C++
PyTorch
JAX
Linux
+1 more

$70–$85 / hr

Posted Sep 21, 2026

Research, Mid Training

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 21, 2026

Senior Research Scientist, Magnets

SandboxAQUnited States

Senior
Remote
Research
3+ years

Python
PyTorch
JAX

$134K–$252K / yr

Posted Sep 21, 2026

F

Machine Learning Engineer (f/m/d)

FocusedDarmstadt, DE

Hybrid
ML / AI engineering

Python
C++
PyTorch
JAX
MLflow
+1 more

Pay not posted

Posted Sep 21, 2026

F

Machine Learning Engineer

FocusedAustin, TX

Hybrid
ML / AI engineering

Python
C++
PyTorch
JAX
MLflow
+1 more

Pay not posted

Posted Sep 21, 2026

RA

Research Member of Technical Staff- Training Systems

Rhoda AIMountain View

Research
Member of Technical Staff

PyTorch
CUDA
JAX

$200K–$300K / yr

Posted Sep 18, 2026

Research Intern, Frontier Agents (Summer 2027)

Together AISan Francisco

Intern
Research

Python
PyTorch
JAX
Salesforce

$58–$70 / hr

Posted Sep 18, 2026

S

Member of Technical Staff, Data and RL

SonderNew York City

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

Python
PyTorch
JAX

$250K–$300K / yr

Posted Sep 18, 2026

Machine Learning Engineer, Driver Refinement & Validation, Perception

WaymoMountain View, CA +3

ML / AI engineering
3+ years

Python
C++
PyTorch
JAX

$175K–$215K / yr

Posted Sep 18, 2026

Research Intern, Model Shaping (Summer 2027)

Together AISan Francisco, Amsterdam +1

Intern
Research

PyTorch
JAX
Salesforce

$58–$70 / hr

Posted Sep 18, 2026

Research Intern, Inference (Summer 2027)

Together AISan Francisco

Intern
Research

Python
PyTorch
CUDA
JAX
Salesforce

$58–$70 / hr

Posted Sep 18, 2026

AI Research Intern – Predictive World Model

XPENGSanta Clara, CA +1

Intern
Research

Python
PyTorch
JAX

Pay not posted

Posted Sep 18, 2026

Senior Computer Vision Engineer (Localization)

Niantic SpatialSan Francisco, CA +1

Senior
Hybrid
ML / AI engineering
5+ years

Python
C++
PyTorch
JAX

$189K–$255K / yr

Posted Sep 18, 2026

Research Engineer

AdobeSeattle

Research
3+ years

Python
C++
PyTorch
JAX
TensorFlow

$121K–$239K / yr

Posted Sep 18, 2026

checked 2 days ago

Go to the first page Show the next 40 jobs