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 (2)

Job typeSoftware engineering46ML / AI engineering491Inference & ML systems366Research188Infra, platform & SRE8Data science11Security engineering1Product management2Quant and trading2Solutions engineering7Hardware and robotics4

LocationBay Area365New York65Remote (US)39Other places in the US97Outside the US52India3Bengaluru1

LevelIntern50Entry level9Senior93Staff54Principal11Director+7

Work typeOnsite128Hybrid50Remote38

Minimum base payNo minimum$100K+429$150K+425$200K+403$250K+324$300K+210

Date postedAll datesLast 24 hours0Last 7 days21Last 30 days132

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

506 jobs from 7,835 companies, checked on their job boards 5 h ago

Sort

Posted pay×JAX×Remove all filters

Email me new jobs like this

Tech jobs. Only jobs with posted pay. 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.

Senior Deep Learning Engineer, Cosmo 3D Spatial
NVIDIAUS, CA, Santa Clara
- Senior
- ML / AI engineering
- 12+ years
- Python
- PyTorch
- JAX
- Linux
$224K–$357K / yr
Posted 16 days ago
checked 5 days ago
PhD Research Intern, Robotics - Summer 2027
NVIDIAUS, WA, Seattle +1
- Intern
- Research
- Python
- C++
- PyTorch
- CUDA
- JAX
- +1 more
$30–$94 / hr
Posted 16 days ago
checked 5 days ago
PhD Research Intern, Quantum Simulation and AI - 2027
NVIDIA

US, CA, Santa Clara

Intern
Research
1+ years

Python
PyTorch
CUDA
JAX
TensorFlow

$38–$94 / hr

Posted 16 days ago

checked 5 days ago

Software Engineer, Performance Instrumentation

NVIDIAUS, CA, Santa Clara +1

Software engineering
8+ years

Python
C++
PyTorch
CUDA
JAX

$168K–$270K / yr

Posted 16 days ago

checked 5 days ago

Robotics Embedded Software Tech Lead, DeepMind

GoogleMountain View, California

Senior
Onsite
Software engineering
8+ years

PyTorch
CUDA
JAX

$207K–$300K / yr

Posted 16 days ago

checked 9 days ago

Senior Research Scientist, Beam

GoogleMountain View, California +2

Senior
Onsite
Research
2+ years
2 positions

Python
JAX

$174K–$252K / yr

Posted 16 days ago

checked 9 days ago

TR

Human Interactive Driving Research Intern, Decision-Making for Driving at the Limits

Toyota Research InstituteLos Altos, CA

Intern
Hybrid
Research

Python
C++
PyTorch
JAX
ROS

$40–$65 / hr

Posted 16 days ago

TR

Human Interactive Driving Research Intern, Perception for Driving at the Limits

Toyota Research InstituteLos Altos, CA

Intern
Hybrid
Research

Python
C++
PyTorch
JAX
ROS

$45–$65 / hr

Posted 16 days ago

Senior Software Engineer, On-Device Machine Learning

GoogleSunnyvale, California

Senior
Onsite
ML / AI engineering
5+ years

PyTorch
JAX
TensorFlow

$174K–$252K / yr

Posted 17 days ago

checked 9 days ago

Senior Software Engineer, AI/ML, Youtube Ads Creative Optimization

GoogleMountain View, California

Senior
Onsite
ML / AI engineering
5+ years

JAX
TensorFlow

$174K–$252K / yr

Posted 17 days ago

checked 9 days ago

Senior Robotics Engineer, Robot Learning

Senra SystemsCypress, California +1

Senior
Software engineering

Python
C++
PyTorch
JAX
ROS

$200K–$260K / yr

Posted 17 days ago

AI/ML Engineer

RiverlaneDelft, NL +1

ML / AI engineering
2+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

€70K–€90K / yr

Posted 17 days ago

Research Engineer, Gen Media, DeepMind

GoogleMountain View, California

Onsite
Research
5+ years

PyTorch
JAX
TensorFlow

$174K–$252K / yr

Posted 18 days ago

checked 9 days ago

A

Software Engineer - Manipulation

ApptronikSunnyvale, CA +1

Software engineering
3+ years

Python
C++
PyTorch
JAX

$180K–$240K / yr

Posted 18 days ago

Machine Learning Engineer, Driving Foundation Models

42dotSunnyvale

ML / AI engineering

Python
AWS
Kubernetes
GCP
Azure
+8 more

$175K–$250K / yr

Posted 18 days ago

AI Research Engineer

RelariSan Francisco, California

Research

Python
PyTorch
JAX

$125K–$175K / yr

Posted 19 days ago

checked 2 days ago

Machine Learning Scientist - Molecular Force Fields

SandboxAQUnited States

Remote
ML / AI engineering

Python
PyTorch
JAX

$134K–$252K / yr

Posted 19 days ago

Machine Learning Scientist, Generative Molecular Design (Material Discovery)

SandboxAQUnited States

Remote
ML / AI engineering
4+ years

AWS
GCP
PyTorch
JAX
TensorFlow

$134K–$252K / yr

Posted 19 days ago

Machine Learning Scientist – Generative Molecular Design

SandboxAQUnited States

Remote
ML / AI engineering
4+ years

AWS
GCP
PyTorch
JAX
TensorFlow

$134K–$252K / yr

Posted 19 days ago

O

Senior Compiler Engineer

OLIXLondon, UK

Senior
Onsite
Inference & ML systems
5+ years

C++
PyTorch
C
JAX
TensorFlow

£210K / yr

Posted 20 days ago

KS

Photonic IC Design Engineer, Design Automation & Modeling

K2 Space Seattle, WA +1

Hardware and robotics

Python
PyTorch
JAX

$130K–$200K / yr

Posted 22 days ago

Research Scientist, Multimodal Alignment, Safety and Fairness, DeepMind

GoogleMountain View, California +2

Onsite
Research

Python
JAX

$147K–$210K / yr

Posted 22 days ago

checked 9 days ago

S

AI Engineer Intern - Fury Team (Summer 2027)

ScoutSunnyvale, CA

Intern
Onsite
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$45–$65 / hr

Posted 22 days ago

T

Robotics Engineering Intern — Hardware & Software

Intern
Onsite
Software engineering

Python
C++
PyTorch
JAX
ROS

$20–$50 / hr

Posted 22 days ago

2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and Efficiency

WaymoMountain View, CA +1

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted 22 days ago

T

Research Intern

Intern
Onsite
Research

Python
C++
PyTorch
CUDA
JAX

$22–$50 / hr

Posted 22 days ago

Research Scientist, GenAI, DeepMind

GoogleMountain View, California

Onsite
Research

Python
PyTorch
JAX
TensorFlow

$207K–$300K / yr

Posted 23 days ago

checked 9 days ago

Senior Machine Learning Engineer, Scenes, Traffic Controls

WaymoMountain View, California, United States, San Francisco, California +2

Senior
ML / AI engineering
4+ years

Python
SQL
C++
PyTorch
JAX
+1 more

$213K–$263K / yr

Posted 23 days ago

2027 Summer Intern, PhD, Research, Multimodal Foundation Models

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX

$85 / hr

Posted 23 days ago

Compiler Frontend Engineer

Normal Computing CorporationMountain View +1

Onsite
Inference & ML systems

Python
C++
Rust
PyTorch
JAX

$200K–$300K / yr

Posted 24 days ago

Software Engineer

WaymoPERM - N/A +1

Software engineering
5+ years

JAX
TensorFlow

$250K–$263K / yr

Posted 24 days ago

Research Engineer, Gemini Large Scale Pretraining, DeepMind

GoogleLondon, England, United Kingdom +1

Onsite
Research
2+ years

JAX

$147K–$210K / yr

Posted 24 days ago

checked 9 days ago

Staff VLA Engineer

Staff
Software engineering

PyTorch
JAX

$250K–$325K / yr

Posted 24 days ago

Senior Scientist I, Computational Protein Design

Generate BiomedicinesSomerville, MA

Senior
Data science
3+ years

Python
PyTorch
JAX

$172K–$241K / yr

Posted 24 days ago

Principal Data Scientist, User Ecosystem

RedditRemote - United States

Principal
Remote
Data science

PyTorch
JAX

$313K–$365K / yr

Posted 24 days ago

Research Scientist, Science AI, Google Research

GoogleCambridge, Massachusetts

Onsite
Research

Python
C++
PyTorch
JAX
TensorFlow

$147K–$210K / yr

Posted 25 days ago

checked 9 days ago

Research Scientist Intern, Multimodal Learning for Robotics (PhD)

MetaBurlingame, California

Intern
Onsite
Research

Python
PyTorch
JAX

$8K–$13K / mo

Posted 25 days ago

checked 9 days ago

Research Scientist Intern, Robot Foundation Models (PhD)

MetaRedmond, Washington

Intern
Onsite
Research

Python
PyTorch
JAX
ROS

$8K–$13K / mo

Posted 25 days ago

checked 9 days ago

AI Performance Engineer - New Grad

SambaNovaAustin, Texas +3

Entry level
Software engineering

Python
C++
PyTorch
CUDA
JAX
+2 more

$131K–$197K / yr

Posted 25 days ago

G

Graduate Software Engineer, AI/ ML Test Systems 2027

GraphcoreBristol, UK +1

Entry level
ML / AI engineering

Python
C++
PyTorch
JAX
Linux
+1 more

£45K / yr

Posted 26 days ago

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