AI Jobs
AI Jobs
Post a job
Sign up Sign in

Full Time JAX AI Jobs in California

Browse 9 AI jobs in California for Oct 2026 with roles open at Thinking Machines Lab, Discord, and Samsung Semiconductor, Inc. paying $163K 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 Full Time JAX AI Jobs in California
Share:
AI Jobs
JAX Jobs in AI
Full Time JAX AI Jobs in California
AI Engineer jobs Ansible jobs Customer Success Manager jobs Cypress jobs Data Engineer jobs Data Scientist jobs Elasticsearch jobs Engineering Manager jobs Ethernet jobs HR jobs JAX jobs MongoDB jobs NCCL jobs OpenTelemetry jobs Platform Engineer jobs Product Manager jobs Program Manager jobs RDMA jobs React jobs Rust jobs SEO jobs Snowflake jobs Tableau jobs TensorRT-LLM jobs Triton jobs TypeScript jobs
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
C
Research Engineer, Safety & Alignment
Cognition / San Francisco, United States
Python jobs PyTorch jobs JAX jobs
7D
TML
RL Systems - Infra
Thinking Machines Lab / San Francisco, United States
PyTorch jobs JAX jobs RLHF jobs
$425K – $475K 8D
SSI
Staff Engineer, ML Profiling Tools
Samsung Semiconductor, Inc. / San Jose, United States / Staff
TensorFlow jobs JAX jobs Excel jobs
$163K – $253K 13D
TML
Research, Mid Training
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 24D
D
Staff Software Engineer, Machine Learning (Consumer Revenue)
Discord / San Francisco, United States / Remote / Staff
Python jobs PyTorch jobs TensorFlow jobs
$272K – $340K 24D
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 Full Time JAX jobs?

9 active listings have salaries ranging from $163K to $530K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include Thinking Machines Lab, Discord, and Samsung Semiconductor, Inc..

Are there remote Full Time JAX jobs?

Yes, 1 of 9 are remote.