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 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 Engineering GenAI - US
MetaMenlo Park, California
- Onsite
- Software engineering
- 5+ years
- Python
- Java
- Go
- C++
- Rust
- +4 more
$212K–$250K / yr
Posted Sep 18, 2026
checked 9 days ago
Staff Tech Lead Manager, Machine Learning Engineering, Simulator Evaluation
WaymoMountain View, California +2
- Staff
- ML / AI engineering
- Manager
- 7+ years
- Python
- C++
- PyTorch
- JAX
- TensorFlow
$251K–$310K / yr
Posted Sep 17, 2026
Sr. Machine Learning Engineer, Siri Speech
AppleCupertino, California

Senior

Onsite

ML / AI engineering

Python
PyTorch
JAX

Pay not posted

Posted Sep 17, 2026

checked 9 days ago

Member of Technical Staff, AI Agent Development Lead

PostmanSan Francisco, CA

Senior
ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX
TensorFlow

$256K–$276K / yr

Posted Sep 17, 2026

checked 2 days ago

Research, Finetuning Science

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 16, 2026

Research Lead, Tinker, Fine-tuning Science

Thinking Machines LabSan Francisco

Senior
Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 16, 2026

W

Staff Machine Learning Engineer

WatneySan Francisco

Staff
Onsite
ML / AI engineering

Python
Kubernetes
PyTorch
JAX

Pay not posted

Posted Sep 16, 2026

2027 Summer Intern, PhD, Machine Learning, Computer Vision

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted Sep 16, 2026

Director, Data Engineering

VisaIN - Bengaluru, India

Director+
Software engineering
Manager
13+ years

AWS
GCP
Azure
PyTorch
Spark
+5 more

Pay not posted

Posted Sep 16, 2026

checked 2 days ago

I

Inference Engineering, Co-op

InferactSan Francisco

Intern
Onsite
Inference & ML systems

Python
Kubernetes
Go
C++
Rust
+6 more

Pay not posted

Posted Sep 15, 2026

Machine Learning Engineer, Natural Language Understanding, Proactive

AppleSanta Clara, California

Onsite
ML / AI engineering
4+ years

Python
Go
C++
PyTorch
Swift
+4 more

Pay not posted

Posted Sep 15, 2026

checked 9 days ago

AA

GPU Kernel Engineer – CUDA, Triton & Accelerator Performance

Anyone AIArgentina - Fully Remote +8

Remote
Inference & ML systems
3+ years

AWS
CUDA
JAX

$65 / hr

Posted Sep 15, 2026

MI

Data Infra - Signals and Evaluations (IC)

Matter IntelligenceSan Francisco

Onsite
Infra, platform & SRE

Python
PyTorch
JAX

Pay not posted

Posted Sep 15, 2026

Cybersecurity Engineer

VisaUS - Foster City, CA

Security engineering
5+ years

Python
AWS
SQL
Kubernetes
GCP
+12 more

$137K–$213K / yr

Posted Sep 15, 2026

checked 2 days ago

Machine Learning Engineer – Ad Sciences

InMobi San Mateo, CA +1

ML / AI engineering
3+ years

JAX

$173K–$210K / yr

Posted Sep 14, 2026

Worldwide Specialist Solutions Architect - GenAI, Data & AI GTM

Amazon Web Services, Inc.Austin, Texas +3

Solutions engineering

AWS
Kubernetes
PyTorch
JAX

$154K–$239K / yr

Posted Sep 14, 2026

checked 2 days ago

RA

Software Engineer, Distributed Training

River AI Inc.Palo Alto, CA +1

Inference & ML systems
Visa sponsorship in the job text

Python
C++
Rust
PyTorch
CUDA
+1 more

$200K–$420K / yr

Posted Sep 12, 2026

Member of Technical Staff - Machine Learning

Inductive BioNew York City, San Francisco, or Boston

Hybrid
ML / AI engineering
Member of Technical Staff
4+ years

PyTorch
JAX
TensorFlow

$180K–$250K / yr

Posted Sep 11, 2026

Machine Learning Engineer - Video Generation Models

AppleCalifornia

Onsite
ML / AI engineering
3+ years

Python
PyTorch
JAX

Pay not posted

Posted Sep 10, 2026

checked 9 days ago

BR

2027 Internship Onboard Infrastructure Engineer, ML Inference

Bedrock Robotics IncSan Francisco, CA

Intern
Onsite
Inference & ML systems

C++
Rust
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted Sep 10, 2026

Senior Machine Learning Engineer, Predictive Planner, Systems Arch

WaymoMountain View, CA +3

Senior
ML / AI engineering
4+ years

Python
C++
PyTorch
C
JAX
+1 more

$213K–$263K / yr

Posted Sep 10, 2026

2027 Summer Intern, PhD, Vehicle Intent and Prediction

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted Sep 10, 2026

Compute Solution Architect

Mistral AINew York +2

Remote
Solutions engineering
5+ years

Kubernetes
PyTorch
JAX
TensorFlow
vLLM

Pay not posted

Posted Sep 10, 2026

AIML - Senior Machine Learning Research Engineer, LLM Post-training (Multilinguality)

AppleCupertino, California +1

Senior
Onsite
Research
2+ years
2 positions

Python
PyTorch
Bash
Hugging Face
JAX
+1 more

Pay not posted

Posted Sep 9, 2026

checked 9 days ago

Staff+ Software Engineer, ML Inference Path

AnthropicSan Francisco, CA

Staff
Inference & ML systems
5+ years
Visa sponsorship in the job text

Python
PyTorch
A/B testing
JAX
TensorFlow

$320K–$485K / yr

Posted Sep 9, 2026

TA

Machine Learning Engineer (II-III), Space Edge Deployment

True AnomalyDenver, CO +1

ML / AI engineering
2+ years

Python
PyTorch
JAX
TensorFlow

$125K–$220K / yr

Posted Sep 9, 2026

Research Software Engineer, Post Training

Thinking Machines LabSan Francisco

Onsite
Research
Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$350K–$475K / yr

Posted Sep 9, 2026

E

Thermo ML Resident

ExtropicSan Francisco +1

Entry level
Onsite
ML / AI engineering
2 positions

Python
Kubernetes
PyTorch
JAX
Ray
+1 more

$75K–$200K / yr

Posted Sep 8, 2026

Research Scientist, Map Scalability

WaymoMountain View, CA, U.S.A +5

Research
1+ years

Python
C++
PyTorch
JAX
TensorFlow

$213K–$263K / yr

Posted Sep 8, 2026

FP

Pioneering Intelligence: Data Science Co-Op (Embedded Science Team)

Flagship Pioneering Co-Op ProgramCambridge, MA USA +1

Intern
Data science

Python
AWS
GCP
Azure
PyTorch
+1 more

$25–$35 / hr

Posted Sep 8, 2026

Software Engineer, GenAI Frameworks

MetaMenlo Park, California

Onsite
Software engineering
2+ years

Python
Java
C++
PyTorch
Spark
+4 more

$122K–$181K / yr

Posted Sep 8, 2026

checked 9 days ago

Founding Research Scientist, Biological Foundation Models

CellTypeNew York, New York

Research

Python
PyTorch
JAX

$150K–$250K / yr

Posted Sep 7, 2026

checked 2 days ago

TR

Senior Researcher, Robot Learning & Foundation Models

The Robotics CompanySan Francisco +1

Senior
Research

PyTorch
JAX

Pay not posted

Posted Sep 7, 2026

C

Research Intern

CantinaSingapore

Intern
Hybrid
Research

Python
PyTorch
JAX

Pay not posted

Posted Sep 7, 2026

Robotics Autonomy Engineer, Manipulation

FieldAIIrvine, CA +1

Onsite
ML / AI engineering
3+ years

Python
C++
PyTorch
JAX
ROS
+1 more

$100K–$200K / yr

Posted Sep 5, 2026

ML Engineer

MonarchEmeryville, California

Onsite
ML / AI engineering

Python
GCP
PyTorch
JAX
TensorFlow

Pay not posted

Posted Sep 5, 2026

Applied Scientist, Neuron ARG, Annapurna ML

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

Research
4+ years

Python
Java
C++
PyTorch
JAX

$143K–$222K / yr

Posted Sep 5, 2026

checked 2 days ago

Computational Scientist, Differentiable Physics

Periodic LabsMenlo Park, CA

Onsite
Data science
Visa sponsorship in the job text

PyTorch
JAX

$250K–$350K / yr

Posted Sep 4, 2026

2027 PhD Residency, Earth Scientist AI Resident (Early Stage Project)

XMountain View, CA +1

Entry level
Data science

Python
JAX
NumPy
scikit-learn
TensorFlow

$109K–$157K / yr

Posted Sep 4, 2026

Research Engineer, Forge

Mistral AIParis +4

Hybrid
Research

Python
Kubernetes
PyTorch
JAX
Ray

Pay not posted

Posted Sep 4, 2026

Go to the first page Show the next 40 jobs