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 (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,867 companies, checked on their job boards 13 min 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.

ML Research Engineer
MapleNew York, NY (HQ)
- Onsite
- Research
- 3+ years
- PyTorch
- CUDA
- JAX
$120K–$250K / yr
Posted Apr 13, 2025
checked 3 days ago
Sr. Machine Learning Engineer, Recommendation Systems
xAIPalo Alto, CA
- Senior
- ML / AI engineering
- 5+ years
- PyTorch
- ClickHouse
- CUDA
- JAX
- Kafka
$180K–$330K / yr
Posted Apr 11, 2025
Research Engineer, Machine Learning (Reinforcement Learning)
AnthropicSan Francisco, CA +1
- Research

Visa sponsorship in the job text

Python
PyTorch
JAX
TensorFlow

$500K–$850K / yr

Posted Apr 1, 2025

Staff/Sr. Staff Machine Learning Engineer - World Modeling, Simulation

WaymoMountain View, CA +1

Staff
ML / AI engineering
8+ years

Python
PyTorch
JAX

$251K–$310K / yr

Posted Mar 6, 2025

ML Runtime Optimization Engineer

Applied IntuitionSunnyvale

Onsite
ML / AI engineering
3+ years

PyTorch
JAX

$215K–$315K / yr

Posted Mar 5, 2025

Member of Technical Staff, Training Infra Engineer

CohereParis +5

Remote
Inference & ML systems
Member of Technical Staff

Python
Kubernetes
PyTorch
JAX

CA$250K–CA$535K / yr

Posted Feb 20, 2025

Member of Technical Staff, Training Performance Engineer

CohereLondon +4

Hybrid
Inference & ML systems
Member of Technical Staff

Python
PyTorch
CUDA
JAX

CA$250K–CA$535K / yr

Posted Feb 20, 2025

Machine Learning Lead, Bellwether

XMountain View, CA (HQ)

Senior
ML / AI engineering
7+ years

Python
PyTorch
JAX
NumPy
pandas
+1 more

$262K–$369K / yr

Posted Feb 1, 2025

MR

Robotics AI Research Engineer

Maven RoboticsSanta Clara, CA +1

Research

Python
PyTorch
JAX

$175K–$350K / yr

Posted Jan 9, 2025

Senior Member of Technical Staff, Multimodal AI

CohereSan Francisco +4

Senior
Remote
ML / AI engineering
Member of Technical Staff

Python
PyTorch
CUDA
JAX
TensorFlow

CA$250K–CA$535K / yr

Posted Dec 3, 2024

Research Engineer - Machine Learning

True3DNew York, New York

Research
7+ years

Python
Kubernetes
C++
PyTorch
CUDA
+1 more

$100K–$250K / yr

Posted Nov 19, 2024

checked 2 days ago

SA

Research Engineer

Skild AIPittsburgh, San Mateo +2

Research

Python
PyTorch
JAX
TensorFlow

$100K–$300K / yr

Posted Nov 14, 2024

SA

Deep Learning Manipulation Engineer

Skild AIPittsburgh, San Mateo +2

ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$100K–$300K / yr

Posted Nov 14, 2024

CS

Audio Applied Research Science Intern

Careershub ShureNiles, IL

Intern
Research

Python
PyTorch
JAX
TensorFlow

$23–$43 / hr

Posted Nov 10, 2024

checked 2 days ago

CS

Audio Applied Research Science Intern

Careersus ShureNiles, IL

Intern
Research

Python
PyTorch
JAX
TensorFlow

$23–$43 / hr

Posted Nov 10, 2024

checked 2 days ago

Member of Technical Staff, Modeling

CohereLondon +5

Remote
Inference & ML systems
Member of Technical Staff

Python
CUDA
JAX
TensorFlow

CA$250K–CA$535K / yr

Posted Nov 1, 2024

Research Engineer, Data Foundations

RunwayRemote

Research
4+ years

Kubernetes
PyTorch
JAX
Ray

$270K–$370K / yr

Posted Oct 10, 2024

SA

Robotics Engineer, Manipulation

Skild AISan Mateo, CA +3

Software engineering

Python
C++
PyTorch
JAX
ROS
+1 more

$100K–$300K / yr

Posted Sep 23, 2024

SA

Software Engineer, AI Training and Infrastructure

Skild AISan Mateo, CA +3

Infra, platform & SRE

Python
AWS
GCP
Azure
C++
+3 more

$100K–$300K / yr

Posted Sep 23, 2024

SA

Research Scientist

Skild AIPittsburgh, San Mateo +2

Research

Python
PyTorch
JAX
TensorFlow

$100K–$300K / yr

Posted Sep 19, 2024

SA

Machine Learning Engineer, Reinforcement Learning

Skild AIPittsburgh, San Mateo +2

ML / AI engineering

Python
C++
PyTorch
JAX
TensorFlow

$100K–$300K / yr

Posted Sep 11, 2024

SA

Research Engineer, Post-training & Deployment

Skild AISan Mateo, CA +1

Research

Python
PyTorch
JAX
ROS
TensorFlow

$100K–$300K / yr

Posted Aug 27, 2024

Machine Learning Scientist

Blue Rose ResearchRemote

ML / AI engineering

Python
SQL
JAX

$140K–$190K / yr

Posted Oct 25, 2023

N

Machine Learning Engineer

NeuralinkAustin, Texas +3

ML / AI engineering

Python
TypeScript
Rust
PyTorch
Swift
+1 more

$129K–$331K / yr

Posted Jun 29, 2023

AI Researcher (Anime)

SpellbrushSan Francisco, California +2

Remote
Research
Visa sponsorship in the job text

PyTorch
JAX

$75K–$200K / yr

Posted Apr 25, 2023

checked 2 days ago

Applied AI Data Scientist

AE StudioLA office / US remote +2

Remote
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

$180K–$240K / yr

Posted Apr 12, 2021

Go to the first page