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

Research Scientist, Multimodal Generative AI (Intelligent Creation)– Global Frontier Tech Recruitment Program -2027 Start (PhD)
TikTokSan Jose, California
- Research
- PyTorch
- JAX
Pay not posted
checked 2 days ago
Machine Learning Engineer Graduate (E-Commerce Recommendation Mall) - 2027 Start (PhD)
TikTokSeattle, Washington
- Entry level
- ML / AI engineering
- Python
- PyTorch
- JAX
- TensorFlow
Pay not posted
checked 2 days ago
Machine Learning Engineer Intern (E-Commerce Supply Chain & Logistics-LLM/Agent) - 2027 Start (PhD)
TikTokSan Jose, California +1
- Intern
- ML / AI engineering

2 positions

Python
TypeScript
Java
Go
C++
+6 more

Pay not posted

checked 2 days ago

Research Scientist/Software Engineer, Neural Graphics and World Models - Global Frontier Tech Recruitment Program - 2027 Start (PhD)

TikTokSan Jose, California

Research

PyTorch
JAX

Pay not posted

checked 2 days ago

Multimodal Generative AI Research Scientist (Intelligent Creation) – Global Frontier Tech Recruitment Program - 2027 Start (PhD)

TikTokSeattle, Washington

Research

PyTorch
JAX

Pay not posted

checked 2 days ago

Neural Graphics and World Models Research Scientist/Software Engineer Graduate (TikTok, Engine&Tools) - 2027 Start (PhD)

TikTokSan Jose, California

Entry level
Research

PyTorch
JAX

Pay not posted

checked 2 days ago

Machine Learning Engineer Graduate (E-Commerce Recommendation Mall) - 2027 Start

TikTokSan Jose, California +1

Entry level
ML / AI engineering
2 positions

Python
PyTorch
JAX
TensorFlow

Pay not posted

checked 2 days ago

Research Scientist Intern (TikTok-Neural Graphics and World Models) - 2027 Start (PhD)

TikTokSan Jose, California

Intern
Research

PyTorch
JAX

Pay not posted

checked 2 days ago

Machine Learning Engineer Intern (E-Commerce Recommendation Mall) - 2027 Start (PhD)

TikTokSan Jose, California +1

Intern
ML / AI engineering
2 positions

Python
PyTorch
JAX
TensorFlow

Pay not posted

checked 2 days ago

Senior Research Scientist/AI Software Engineer, LLM/Agent Platform (TikTok-Content Ecology AI Innovation & Platform) - TikTok Content Ecology

TikTokSan Jose, California

Senior
Research
5+ years

PyTorch
JAX
LangChain
TensorFlow
vLLM

Pay not posted

checked 2 days ago

Machine Learning Engineer Intern (E-Commerce Supply Chain & Logistics-LLM / Agent )- 2027 Summer

TikTokSeattle, Washington

Intern
ML / AI engineering

Python
TypeScript
Java
Go
C++
+6 more

Pay not posted

checked 2 days ago

Machine Learning Engineer Intern (E-Commerce Recommendation Mall)- 2027 Summer

TikTokSeattle, Washington

Intern
ML / AI engineering

Python
PyTorch
JAX
TensorFlow

Pay not posted

checked 2 days ago

Tech Lead Research Scientist/Software Engineer, TikTok-Neural Graphics and World Models

TikTokSan Jose, California

Senior
Research
5+ years

PyTorch
JAX

Pay not posted

checked 2 days ago

Research Scientist / AI Software Engineer, LLM/Agent Platform - TikTok - Content Ecology AI Innovation & Platform - TikTok Content Ecology

TikTokSan Jose, California

Research
2+ years

PyTorch
JAX
LangChain
TensorFlow
vLLM

Pay not posted

checked 2 days ago

Machine Learning Researcher

D. E. ShawNew York

Research

PyTorch
CUDA
JAX

$275K–$350K / yr

checked 2 days ago

Sr. Model Engineer, Voice

xAIPalo Alto, CA

Senior
Software engineering
2+ years

Python
Kubernetes
C++
Rust
PyTorch
+2 more

$220K–$357K / yr

Posted 8 days ago

S

GPU Kernel Engineer

SciforiumSan Francisco, CA

Onsite
Inference & ML systems
5+ years

Python
C++
PyTorch
CUDA
JAX
+1 more

$190K–$250K / yr

Posted 8 days ago

Senior Forward Deployed Engineer

SukiRedwood City, CA +1

Senior
Software engineering
8+ years

Python
TypeScript
Kubernetes
GCP
React
+7 more

$225K–$250K / yr

Posted 8 days ago

AIML - Sr Machine Learning Platform Engineer, Evaluation

AppleSeattle, Washington

Senior
Onsite
ML / AI engineering
6+ years

Python
PyTorch
JAX

Pay not posted

Posted 8 days ago

checked 9 days ago

2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable Simulation

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$85 / hr

Posted 8 days ago

2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)

WaymoMountain View, California +1

Intern
ML / AI engineering

Python
PyTorch
JAX

$70–$85 / hr

Posted 8 days ago

CR

Research Software Engineer

Convergent ResearchBoulder, CO

Onsite
Research
3+ years

Python
PyTorch
CUDA
JAX
NumPy

$100K–$175K / yr

Posted 8 days ago

Sr. Machine Learning Engineer, Autonomy

42dotSunnyvale

Senior
ML / AI engineering
5+ years

Python
PyTorch
JAX

$210K–$300K / yr

Posted 8 days ago

B

Applied AI Engineer

BjakUnited States +23

Remote
ML / AI engineering
3+ years
36 positions

Python
PyTorch
JAX
vLLM

Pay not posted

Posted 9 days ago

Senior System Software Engineer, DGX Cloud AI Infrastructure

NVIDIAUS, CA, Santa Clara

Senior
ML / AI engineering
12+ years

Kubernetes
PyTorch
JAX
Ray
TensorFlow

$224K–$431K / yr

Posted 9 days ago

checked 5 days ago

RA

Applied Researcher

River AI Inc.Palo Alto, CA +1

Research
Visa sponsorship in the job text

PyTorch
JAX

$200K–$420K / yr

Posted 9 days ago

Computer Vision & Machine Learning Engineer

AppleSunnyvale, California

Onsite
ML / AI engineering

Python
PyTorch
JAX

Pay not posted

Posted 9 days ago

checked 9 days ago

Staff Machine Learning Engineer, AI Infrastructure

WaymoMountain View, CA +5

Staff
ML / AI engineering
7+ years

Python
C++
PyTorch
JAX
TensorFlow

$251K–$310K / yr

Posted 9 days ago

T

Member of Technical Staff - Research Intern

TrajectorySan Francisco

Intern
Onsite
Research
Member of Technical Staff

PyTorch
JAX

Pay not posted

Posted 9 days ago

Research Scientist, Discovery (Engineering & Applied Sciences)

AnthropicSan Francisco, CA

Research
Visa sponsorship in the job text

Python
PyTorch
JAX

$500K–$850K / yr

Posted 9 days ago

L

Senior Backend Engineer, Pricing Platform

Senior
Infra, platform & SRE
10+ years

Python
TypeScript
Rust
JAX

Pay not posted

Posted 10 days ago

Senior Software Engineer, ML Performance, GenAI, DeepMind

GoogleMountain View, California

Senior
Onsite
Inference & ML systems
5+ years

PyTorch
JAX
vLLM

$174K–$252K / yr

Posted 10 days ago

checked 9 days ago

CP

AI/Machine Learning Research Scientist – Undersea Acoustics & Physics-Based Modeling

Careers PeratonStennis Space Center, MS

Research
8+ years

PyTorch
JAX
TensorFlow

$104K–$166K / yr

Posted 10 days ago

checked 2 days ago

P

Machine Learning Scientist, Biomolecular Structure Prediction

ProfluentEmeryville, California +1

ML / AI engineering

PyTorch
JAX

$212K–$339K / yr

Posted 10 days ago

Sr. Model Engineer, RL Training Framework

xAIPalo Alto, CA

Senior
ML / AI engineering

Python
C++
Rust
PyTorch
JAX

$220K–$357K / yr

Posted 10 days ago

Senior Software Engineer

LanceDBAmericas or APAC

Senior
Software engineering
10+ years

Java
C++
Rust
PyTorch
Spark
+6 more

$180K–$250K / yr

Posted 10 days ago

Senior Machine Learning Engineer (Growth)

DiscordSan Francisco Bay Area +1

Senior
ML / AI engineering
5+ years

Python
PyTorch
JAX
TensorFlow

$220K–$275K / yr

Posted 10 days ago

Senior Staff Machine Learning Engineer, Perception, Simulation Realism

WaymoMountain View, California +2

Staff
ML / AI engineering

Python
C++
PyTorch
JAX
NumPy

$298K–$368K / yr

Posted 10 days ago

Senior Software Engineer, Machine Learning, Discover Interest Modeling and Retrieval

GoogleMountain View, California

Senior
Onsite
ML / AI engineering
5+ years

Python
SQL
C++
PyTorch
Spark
+2 more

$174K–$252K / yr

Posted 10 days ago

checked 9 days ago

Applied Scientist Lead - Robotics, EMEA

Mistral AIParis +3

Senior
Hybrid
Research

Python
PyTorch
JAX

Pay not posted

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