AI Jobs
AI Jobs
Post a job
Sign up Sign in

JAX Jobs with Paid Time Off in AI

Browse 12 JAX jobs with Paid Time Off in AI for Oct 2026 with roles open at Thinking Machines Lab, The Allen Institute for AI, and Samsung Semiconductor, Inc. paying $146K to $530K/year. Work as a Research, Tinker, RL Systems, Research Lead, Tinker, Fine-tuning Science, and Research, Finetuning Science.

JAX is often requested for Software Engineer jobs roles.

Receive emails of JAX Jobs with Paid Time Off in AI
Share:
AI Jobs
JAX Jobs in AI
JAX Jobs with Paid Time Off in AI
AI Engineer jobs Ansible jobs Communications Lead jobs Customer Success Manager jobs Cypress jobs Data Engineer jobs Data Scientist jobs Elasticsearch jobs Engineering Manager jobs Ethernet jobs Full Stack Engineer jobs HR jobs JAX jobs MongoDB jobs NCCL jobs OpenTelemetry jobs Product Manager jobs Program Manager jobs RDMA jobs React jobs Rust jobs SEO jobs Tableau jobs TensorRT-LLM jobs Triton jobs TypeScript jobs
Filter by benefits
🚑 Medical Insurance 🦷 Dental Insurance 🤓 Vision Insurance 💰 401k 💰 401k Matching 💰 Equity 🏖 Unlimited Vacation 🏖 Paid Time Off 🏔 Company Retreats 🖥 Home Office Budget 📚 Learning Budget 🧘 Mental Wellness Budget ✈️ Visa Sponsorship 📦 Relocation 💊 FSA 🏥 HSA
Filter by type
Remote Full Time Internship
C
Senior Staff Software Engineer, Storage
Crusoe / Sunnyvale, United States / Staff
PyTorch jobs TensorFlow jobs JAX jobs
$245K – $290K 1D
TML
RL Systems - Infra
Thinking Machines Lab / San Francisco, United States
PyTorch jobs JAX jobs RLHF jobs
$425K – $475K 8D
CI
Software Engineer, Numerical Computing
CHAOS Industries / Washington, United States
Python jobs PyTorch jobs JAX jobs
$150K – $210K 9D
PLI
Senior Software Engineer
Planet Labs Inc. / Ljubljana, Slovenia / Senior
Python jobs JAX jobs GCP jobs
€58K – €72K 9D
SSI
Staff Engineer, ML Profiling Tools
Samsung Semiconductor, Inc. / San Jose, United States / Staff
TensorFlow jobs JAX jobs Excel jobs
$163K – $253K 14D
TAI
Research Engineer, Climate Modeling
The Allen Institute for AI / Seattle, United States
Python jobs JAX jobs
$147K – $220K 21D
TAI
Research Scientist, Climate Modeling
The Allen Institute for AI / Seattle, United States
Python jobs JAX jobs
$167K – $261K 21D
TAI
Young Investigator, Climate Modeling
The Allen Institute for AI / Seattle, United States
JAX jobs
$160K+ 21D
TML
Research, Mid Training
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 25D
TML
Research, Finetuning Science
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 29D
TML
Research Lead, Tinker, Fine-tuning Science
Thinking Machines Lab / San Francisco, United States / Hybrid / Senior
PyTorch jobs TensorFlow jobs JAX jobs
$475K – $530K 29D
TML
Research, Tinker, RL Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
Python jobs PyTorch jobs TensorFlow jobs
$350K – $475K 29D

Frequently asked questions

What is the average salary for JAX jobs with Paid Time Off in AI?

12 active listings have salaries ranging from $146K to $530K/year.

Which companies are hiring for JAX jobs with Paid Time Off in AI?

Top companies hiring include Thinking Machines Lab, The Allen Institute for AI, and Samsung Semiconductor, Inc..