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

Staff/ Principal Software Engineer (L6/L7): AI Platform Model Serving
NetflixRemote
- Principal
- Onsite
- Inference & ML systems
- PyTorch
- CUDA
- Hugging Face
- JAX
- TensorFlow
- +1 more
Pay not posted
Posted 19 days ago
checked 2 days ago
PL
Research Scientist, AI
Pramaana LabsPalo Alto +1
- Onsite
- Research
- Python
- PyTorch
- JAX
Pay not posted
Posted 19 days ago
AI Research Engineer
RelariSan Francisco, California
- Research
- Python
- PyTorch
- JAX

$125K–$175K / yr

Posted 19 days ago

checked 2 days ago

JC

2027 Machine Learning Center of Excellence (Quant AI) - Summer Associate

JPMorgan Chase & Co.Palo Alto, California

Onsite
ML / AI engineering

Python
Java
C++
PyTorch
C
+2 more

Pay not posted

Posted 19 days ago

checked 2 days ago

Machine Learning Engineer, Answers Knowledge and Information - Proactive

AppleSeattle, Washington +1

Onsite
ML / AI engineering
10+ years
2 positions

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted 19 days ago

checked 9 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

AIML - ML Researcher, AFM

AppleCupertino, California

Onsite
Research

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted 19 days ago

checked 9 days ago

Autonomy Engineer Intern, Reinforcement Learning, 2027 (Zurich)

SkydioZurich, Switzerland

Intern
Onsite
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted 19 days ago

Autonomy Engineer Intern, Reinforcement Learning, 2027 (HQ)

SkydioSan Mateo, California

Intern
Onsite
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+1 more

Pay not posted

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

GPU Architecture and AI Performance Simulation Engineer

IntelIndia, Bangalore

Inference & ML systems
10+ years

PyTorch
CUDA
JAX
TensorFlow

Pay not posted

Posted 20 days ago

checked 2 days ago

Member of Technical Staff - ML Systems

NextTokenMountain View +1

Onsite
Inference & ML systems
Member of Technical Staff

PyTorch
CUDA
JAX
vLLM

Pay not posted

Posted 21 days ago

Member of Technical Staff - Research Engineering

NextTokenMountain View +1

Onsite
Research
Member of Technical Staff

PyTorch
JAX

Pay not posted

Posted 21 days ago

Sr. Computer Vision and Machine Learning Engineer, Creativity Apps

AppleCupertino, California +1

Senior
Onsite
ML / AI engineering
4+ years

Python
C++
PyTorch
Swift
JAX
+1 more

Pay not posted

Posted 22 days ago

checked 9 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

Member of Technical Staff, AI Engineering for Biosecurity

Radical NumericsSan Francisco

ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX

Pay not posted

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

C

Platform Engineer

CuspAILondon, UK +3

Hybrid
Infra, platform & SRE

Python
AWS
Kubernetes
GCP
Go
+5 more

Pay not posted

Posted 22 days ago

T

Research Intern

Intern
Onsite
Research

Python
C++
PyTorch
CUDA
JAX

$22–$50 / hr

Posted 22 days ago

Multimodal LLMs Research Engineer

AppleSunnyvale, California

Onsite
Research
7+ years
2 positions

Python
PyTorch
JAX

Pay not posted

Posted 22 days ago

checked 9 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

S

Member of Technical Staff, Robotics Research

SieveSan Francisco

Onsite
Research
Member of Technical Staff
4+ years

Python
PyTorch
JAX

Pay not posted

Posted 23 days ago

2027 Summer Intern, Onboard Autonomy, Embodied AI (PhD)

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
C++
PyTorch
JAX
ROS
+1 more

Pay not posted

Posted 23 days ago

checked 2 days ago

2027 Summer Intern, Onboard Autonomy, Embodied AI (Masters)

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
C++
PyTorch
JAX
ROS
+1 more

Pay not posted

Posted 23 days ago

checked 2 days ago

EV

Member of Technical Staff — Agent Post-Training

Embedding VCSan Francisco Bay Area

Onsite
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX

Pay not posted

Posted 23 days ago

EV

Member of Technical Staff — RL & Agentic Model Training

Embedding VCSan Francisco Bay Area

Onsite
ML / AI engineering
Member of Technical Staff

Python
PyTorch
JAX

Pay not posted

Posted 23 days ago

EV

Member of Technical Staff — Post-Training Infrastructure

Embedding VCSan Francisco Bay Area

Onsite
Inference & ML systems
Member of Technical Staff

Python
PyTorch
JAX

Pay not posted

Posted 23 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

Member of Technical Staff - Pre-Training Data

ReflectionSan Francisco, CA

Onsite
Inference & ML systems
Member of Technical Staff
Visa sponsorship in the job text

Python
PyTorch
JAX

Pay not posted

Posted 23 days ago

AC

AI Research Intern (PhD)

ATOMS Careers pageSan Francisco, CA +1

Intern
Research

Python
C++
PyTorch
JAX
Ray

Pay not posted

Posted 23 days ago

AI Research Intern (PhD)

AtomsSan Francisco, CA +1

Intern
Research

Python
C++
PyTorch
JAX
Ray

Pay not posted

Posted 23 days ago

Lead Data Scientist

SliceBengaluru, Karnataka, India

Senior
Onsite
Data science

Python
Kubernetes
Docker
PyTorch
JAX
+2 more

Pay not posted

Posted 24 days ago

checked 2 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

IR

Robotics Machine Learning Engineer

Ixian RoboticsSan Francisco

Onsite
ML / AI engineering

Python
PyTorch
JAX

Pay not posted

Posted 24 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