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.

Applied Scientist - Optimisation, EMEA
Mistral AIParis +3
- Research
- Python
- PyTorch
- JAX
Pay not posted
Posted 11 days ago
Lead ML Engineer
Ooak DataParis, Île-de-France, FR
- Senior
- ML / AI engineering
- 8+ years
- Python
- PyTorch
- JAX
€90K–€120K / yr
Posted 11 days ago
checked 2 days ago
Applied Scientist - Tabular Data, EMEA
Mistral AIParis +3
- Hybrid
- Research
- Python
- SQL
- PyTorch
- JAX
- pandas
Pay not posted

Posted 11 days ago

Senior Staff Software Engineer, AI/ML, Google Cloud Compute Infrastructure

GoogleSunnyvale, California

Staff
Onsite
ML / AI engineering
8+ years

GCP
PyTorch
JAX
TensorFlow

$262K–$364K / yr

Posted 11 days ago

checked 9 days ago

Robotics, Hardware Engineer, Simulation Engineer

Lumen LabsSan Francisco, CA +1

Onsite
Hardware and robotics

Python
JAX
ROS

$150k–200k + equity

Posted 11 days ago

checked 2 days ago

Senior CV/ML Generalist Engineer

AppleSunnyvale, California

Senior
Onsite
ML / AI engineering
5+ years

Python
C++
PyTorch
JAX

Pay not posted

Posted 11 days ago

checked 9 days ago

CV/ML Generalist Engineer

AppleSunnyvale, California

Onsite
ML / AI engineering
3+ years

Python
C++
PyTorch
JAX

Pay not posted

Posted 11 days ago

checked 9 days ago

SS

Staff Engineer, Compiler

Samsung SemiconductorSan Jose, California +1

Staff
Inference & ML systems
10+ years

PyTorch
JAX
TensorFlow

$163K–$253K / yr

Posted 11 days ago

PRINCIPAL, DATA SCIENTIST

WalmartIN KA BANGALORE Home Office PTPP1 +2

Principal
Data science
10+ years
3 positions

Python
PyTorch
Spark
A/B testing
BigQuery
+3 more

Pay not posted

Posted 11 days ago

checked 2 days ago

R

Machine Learning Engineer Intern, AI & Robotics

RoboForceMilpitas, CA

Intern
ML / AI engineering

PyTorch
JAX
TensorFlow

$5K–$7K / mo

Posted 11 days ago

Quantum Characterization, Verification, and Validation Engineer

PsiQuantumMilpitas, California

Software engineering

Python
PyTorch
CUDA
JAX
NumPy
+1 more

$89K–$149K / yr

Posted 11 days ago

Senior Staff/Principal Engineer, Foundation Model Compute Infra

AppleCupertino, California +2

Principal
Onsite
Infra, platform & SRE
8+ years

Python
Go
C++
JAX

Pay not posted

Posted 11 days ago

checked 9 days ago

Machine Learning Engineer - New Grad

SambaNovaAustin, Texas +3

Entry level
ML / AI engineering

Python
PyTorch
JAX
TensorFlow
vLLM

$131K–$197K / yr

Posted 11 days ago

Machine Learning Engineer, Siri Speech

AppleCupertino, California

Onsite
ML / AI engineering

Python
PyTorch
JAX

Pay not posted

Posted 11 days ago

checked 9 days ago

Staff Software Engineer, Server and Full Stack, Growth and Discovery, DeepMind

GoogleNew York, New York

Staff
Onsite
Software engineering
8+ years

Python
Java
Go
C++
PyTorch
+2 more

$207K–$300K / yr

Posted 11 days ago

checked 9 days ago

Software Engineering Manager, GenAI, Computer Vision, Google Maps Immersive Navigation

GoogleMountain View, California

Onsite
ML / AI engineering
Manager
8+ years

Python
Java
C++
JavaScript
C
+2 more

$207K–$300K / yr

Posted 12 days ago

checked 9 days ago

E

Research Scientist – Multimodal Foundation Models & Visual Reasoning

ElorianPalo Alto

Onsite
Research

Python
PyTorch
JAX

Pay not posted

Posted 12 days ago

Engineering Manager - ML/AI Products

XeroAU: Sydney (45 Clarence St)

Hybrid
ML / AI engineering
Manager

JAX

Pay not posted

Posted 12 days ago

Senior Applied Research Scientist – AI Native Numerical Methods

NVIDIAUS, CA, Remote +4

Senior
Research
5+ years

Python
PyTorch
JAX

$192K–$357K / yr

Posted 12 days ago

checked 5 days ago

Senior Software Development Engineer, AWS Kubernetes (K8s)

Amazon Development Center U.S., Inc.Santa Clara, California

Senior
Infra, platform & SRE

AWS
Kubernetes
PyTorch
JAX

$193K–$262K / yr

Posted 12 days ago

checked 2 days ago

Machine Learning Engineer, Bellwether, X

XMountain View, CA (HQ)

ML / AI engineering
5+ years

Python
Kubernetes
Docker
Terraform
PyTorch
+4 more

$174K–$261K / yr

Posted 12 days ago

Staff Machine Learning Engineer, Sensors & Intelligence

TonalToronto, ON

Staff
Remote
ML / AI engineering

PyTorch
JAX
TensorFlow

CA$160K–CA$185K / yr

Posted 12 days ago

Research Engineer, Infrastructure

HarveyNew York +3

Hybrid
Research
4 positions

Python
PyTorch
JAX

$230K–$450K / yr

Posted 12 days ago

Sr. Model Engineer, Post-Training & RL

xAIPalo Alto, CA

Senior
Inference & ML systems

Python
C++
Rust
PyTorch
JAX

$220K–$357K / yr

Posted 12 days ago

Sr. Model Engineer, Mid-Training

xAIPalo Alto, CA

Senior
Software engineering

Python
Docker
C++
Rust
PyTorch
+3 more

$220K–$357K / yr

Posted 12 days ago

Senior Software Engineer, AI/ML, Geo and Gemini App

GoogleNew York, New York

Senior
Onsite
ML / AI engineering
5+ years

C++
PyTorch
A/B testing
JAX
TensorFlow

$174K–$252K / yr

Posted 13 days ago

checked 9 days ago

Senior Deep Learning Frameworks CUDA Software Engineer

NVIDIAUS, CA, Santa Clara +2

Senior
Inference & ML systems
8+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$184K–$357K / yr

Posted 13 days ago

checked 5 days ago

Senior Compiler Engineer - AI

NVIDIAUS, CA, Santa Clara +4

Senior
Inference & ML systems
6+ years
2 positions

Python
C++
PyTorch
C
JAX

$184K–$288K / yr

Posted 13 days ago

checked 5 days ago

RA

Senior Inference Optimization ML Engineer

Rhoda AIMountain View

Senior
Inference & ML systems
2+ years

AWS
PyTorch
JAX
vLLM

Pay not posted

Posted 14 days ago

Senior Deep Learning Compiler Engineer - PyTorch

NVIDIAGermany, Berlin +2

Senior
Inference & ML systems
8+ years

Python
PyTorch
JAX

€101K–€194K / yr

Posted 15 days ago

checked 5 days ago

Deep Learning Compiler Verification Engineer

NVIDIAUS, CA, Santa Clara +3

Inference & ML systems
2+ years

Python
C++
PyTorch
JAX

$116K–$190K / yr

Posted 15 days ago

checked 5 days ago

Senior Scientist, Machine Learning Research

A-Alpha BioSeattle, Washington +1

Senior
Research
12+ years

AWS
PyTorch
JAX

$164K–$206K / yr

Posted 15 days ago

Staff Research Scientist, Foundational Audio

GoogleMountain View, California +1

Staff
Onsite
Research
2+ years

PyTorch
Hugging Face
JAX
TensorFlow

$174K–$252K / yr

Posted 15 days ago

checked 9 days ago

AIML - Site Lead & Lead Researcher, Foundation Models

AppleSeattle, Washington

Senior
Onsite
Research

Python
PyTorch
JAX
TensorFlow

Pay not posted

Posted 15 days ago

checked 9 days ago

Senior Staff AI Engineer

JazzX AIBangalore +1

Staff
ML / AI engineering
10+ years

Python
Kubernetes
Java
C++
PyTorch
+3 more

$207K–$290K / yr

Posted 16 days ago

GI

Member of Technical Staff – Infrastructure

General Intuition & MedalNew York City

Onsite
Inference & ML systems
Member of Technical Staff
6+ years

Kubernetes
GCP
C++
Rust
Terraform
+2 more

$300K–$500K / yr

Posted 16 days ago

Solutions Architect, LLM Model Builder

NVIDIAUS, CA, Santa Clara

Solutions engineering
5+ years

Python
PyTorch
CUDA
JAX
TensorFlow
+1 more

$152K–$242K / yr

Posted 16 days ago

checked 5 days ago

Senior Solutions Architect, Robotics Foundation Model Training

NVIDIAUS, CA, Santa Clara +1

Senior
Solutions engineering
5+ years

PyTorch
Hugging Face
JAX

$152K–$288K / yr

Posted 16 days ago

checked 5 days ago

Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIAUS, CA, Santa Clara

Senior
Research
10+ years

Python
Kubernetes
C++
PyTorch
CUDA
+2 more

$224K–$357K / yr

Posted 16 days ago

checked 5 days ago

Senior Robotics Research Scientist

NVIDIAUS, WA, Seattle

Senior
Research
3+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$192K–$357K / yr

Posted 16 days ago

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