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.

TU
AI Software Intern (Canada)
Tenstorrent University JobsToronto, Ontario, Canada +1
- Intern
- Software engineering
- Python
- C++
- PyTorch
- C
- CUDA
- +2 more
Pay not posted
Posted Oct 8, 2026
Director of Product Management, Agentforce Voice Models
SalesforceCalifornia - San Francisco +1
- Director+
- Product management
- Manager
- 10+ years
- Python
- AWS
- GCP
- Azure
- PyTorch
- +1 more
$197K–$345K / yr
Posted Oct 8, 2026
checked 2 days ago
Machine Learning Engineer, Applied
TraceUnited States

+2

Remote
ML / AI engineering
2+ years

Python
PyTorch
JAX

$185K–$245K / yr

Posted Oct 7, 2026

CI

Software Engineer, Numerical Computing

CHAOS IndustriesWashington, District of Columbia +1

Software engineering
5+ years

Python
PyTorch
CUDA
JAX
Jira
+2 more

$150K–$210K / yr

Posted Oct 7, 2026

TU

AI Software Intern (USA)

Tenstorrent University JobsAustin, Texas +3

Intern
Software engineering

Python
C++
PyTorch
C
CUDA
+2 more

$50–$70 / hr

Posted Oct 7, 2026

2027 Summer Intern, MS/PhD, Research, Perception Foundation Models

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX
TensorFlow

$70–$85 / hr

Posted Oct 7, 2026

T

Member of Technical Staff - ML Infra

TrajectorySan Francisco

Onsite
ML / AI engineering
Member of Technical Staff

Python
C++
PyTorch
CUDA
JAX
+1 more

Pay not posted

Posted Oct 6, 2026

PhD Machine Learning Software Engineer Intern (Summer 2027)

LyftSan Francisco, CA +1

Intern
ML / AI engineering

Python
SQL
PyTorch
Spark
A/B testing
+2 more

$65–$68 / hr

Posted Oct 6, 2026

TR

Engineering Intern

The Robotics CompanyZurich +1

Intern
Onsite
Software engineering

Python
C++
PyTorch
JAX
ROS

Pay not posted

Posted Oct 6, 2026

Distinguished Engineer, AI Infrastructure

AppleCupertino, California +1

Principal
Onsite
ML / AI engineering
15+ years

Python
Kubernetes
Go
C++
Rust
+2 more

Pay not posted

Posted Oct 6, 2026

checked 9 days ago

S

Senior HPC & GPU Infrastructure Engineer

SciforiumSan Francisco, CA

Senior
Onsite
Inference & ML systems
5+ years

Python
Kubernetes
Terraform
PyTorch
Ansible
+5 more

$150K–$220K / yr

Posted Oct 6, 2026

Senior Applied Scientist, AWS Neuron Science - Core Algorithm

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

Senior
Research
6+ years

Python
AWS
Java
C++
PyTorch
+6 more

$192K–$260K / yr

Posted Oct 6, 2026

checked 2 days ago

On-Device ML Quality Infrastructure Engineer, Graphics, Games & ML

AppleCupertino, California

Onsite
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted Oct 5, 2026

checked 9 days ago

On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning

AppleCupertino, California

Onsite
ML / AI engineering

Python
C++
PyTorch
Swift
JAX
+1 more

Pay not posted

Posted Oct 5, 2026

checked 9 days ago

IO

Machine Learning Engineer — Reinforcement Learning

Institute of Foundation ModelsSunnyvale, CA

Onsite
ML / AI engineering
5+ years

Python
Kubernetes
PyTorch
CUDA
JAX
+2 more

$150K–$450K / yr

Posted Oct 5, 2026

IO

Machine Learning Engineer — Pre-training (LLM)

Institute of Foundation ModelsSunnyvale, CA

Onsite
Inference & ML systems
5+ years

Python
Kubernetes
PyTorch
CUDA
JAX
+1 more

$150K–$450K / yr

Posted Oct 5, 2026

O

Machine Learning Engineer

OpenMindSan Francisco

Onsite
ML / AI engineering

Python
Go
C++
PyTorch
JAX
+1 more

$130K–$200K / yr

Posted Oct 5, 2026

SoC Machine Learning Design Engineer

AppleCupertino, California +1

Onsite
ML / AI engineering
3+ years
2 positions

PyTorch
JAX
TensorFlow
VHDL

Pay not posted

Posted Oct 5, 2026

checked 9 days ago

Senior Software Engineer, Vehicle Interaction

WaymoMountain View, CA +1

Senior
Software engineering
5+ years

C++
PyTorch
JAX

$213K–$263K / yr

Posted Oct 5, 2026

Senior ML Engineer, Computer Vision

Anduril IndustriesIrvine, California +2

Senior
Remote
ML / AI engineering
6+ years

Python
C++
PyTorch
JAX
Linux
+1 more

$220K–$292K / yr

Posted Oct 5, 2026

Machine Learning Researcher - Agentic Science

PostEraRemote (US)

Remote
Research

Python
PyTorch
JAX

$200K–$300K / yr

Posted Oct 5, 2026

checked 2 days ago

PF

Software Engineer - Simulation Platform

Proxima FusionMunich

Onsite
Infra, platform & SRE

Python
TypeScript
React
FastAPI
JAX

Pay not posted

Posted Oct 5, 2026

Principal, Data Scientist - Applied Machine Learning

Walmart(USA) VIZIO SAN FRANCISCO CA San Francisco Home Office +1

Principal
ML / AI engineering
3+ years

Python
PyTorch
Spark
JAX
TensorFlow

$132K–$286K / yr

Posted Oct 5, 2026

checked 2 days ago

MS

Staff Software Engineer, IR Data Products

Muon SpaceDenver, CO +1

Staff
Remote
Software engineering
10+ years

AWS
Terraform
Airflow
CUDA
JAX
+1 more

$216K–$239K / yr

Posted Oct 2, 2026

SS

Staff Engineer, ML Profiling Tools

Samsung SemiconductorSan Jose, California +1

Staff
ML / AI engineering
10+ years

PyTorch
JAX
TensorFlow

$163K–$253K / yr

Posted Oct 2, 2026

2027 Summer Intern, PhD, Research, World Action Model

WaymoMountain View, CA +1

Intern
Research

Python
PyTorch
JAX
TensorFlow

$85 / hr

Posted Oct 2, 2026

CR

Scientist - AI & Machine Learning

Charles River AnalyticsCambridge, MA

ML / AI engineering
3+ years

Python
Docker
PyTorch
JAX
Linux
+1 more

$120K–$200K / yr

Posted Oct 2, 2026

EH

Software Engineer - ML Infrastructure

Epsilon HealthSan Francisco, CA

Onsite
ML / AI engineering
6+ years

Python
Kubernetes
PyTorch
A/B testing
JAX
+1 more

Pay not posted

Posted Oct 2, 2026

Research Platform Engineer

Tower Research CapitalNew York

Research

Kubernetes
PyTorch
JAX

$200K–$300K / yr

Posted Oct 2, 2026

Data Scientist

MillenniumHong Kong, Hong Kong

Onsite
Data science
3+ years

Python
Kubernetes
Docker
Java
Go
+6 more

Pay not posted

Posted Oct 1, 2026

checked 2 days ago

2027 Summer Intern - AI/ML Engineer, Autonomous Vehicle: Simulation

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
Java
C++
PyTorch
JAX
+1 more

$8K–$10K / mo

Posted Oct 1, 2026

checked 2 days ago

2027 Summer Intern – AI/ML Engineer, Autonomous Vehicles: Simulation (PhD)

General MotorsSunnyvale, California

Intern
Hybrid
ML / AI engineering

Python
C++
PyTorch
CUDA
JAX
+1 more

$12K–$15K / mo

Posted Oct 1, 2026

checked 2 days ago

AI Lab - Research

Optiver New York, New York +1

Research

Python
PyTorch
CUDA
JAX

$200K / yr

Posted Sep 30, 2026

Software Engineer, Evaluation Infrastructure

xAIPalo Alto, CA

Infra, platform & SRE

Python
C++
Rust
PyTorch
JAX

$180K–$440K / yr

Posted Sep 30, 2026

CP

Research Scientist - RF Machine Learning

Careers PeratonCollege Park, MD

Hardware and robotics
10+ years

Python
Java
C++
PyTorch
C
+9 more

$146K–$234K / yr

Posted Sep 30, 2026

checked 2 days ago

Machine Learning Engineer - Proactive

AppleCupertino, California

Onsite
ML / AI engineering

Python
C++
PyTorch
A/B testing
C
+2 more

Pay not posted

Posted Sep 30, 2026

checked 9 days ago

Senior ML Compiler Engineer, Neuron

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

Senior
Inference & ML systems
5+ years

PyTorch
JAX

$168K–$262K / yr

Posted Sep 30, 2026

checked 2 days ago

Research Engineer / Research Scientist, RL Frontiers

AnthropicSan Francisco, CA +2

Research
Visa sponsorship in the job text

Python
C++
Rust
PyTorch
JAX

$500K–$850K / yr

Posted Sep 29, 2026

U

CAD Automation Researcher

UniversalAGISan Francisco

Onsite
Research
2+ years
Visa sponsorship in the job text

Python
PyTorch
JAX

Pay not posted

Posted Sep 29, 2026

2027 Summer Intern, PhD, Planner Machine Learning

WaymoSan Francisco, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted Sep 29, 2026

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