AI Jobs
AI Jobs
Post a job
Sign up Sign in

JAX AI Jobs in San Francisco, California, United States

Browse 8 AI jobs in San Francisco for Oct 2026 with roles open at Together AI, Discord, and Thinking Machines Lab paying $120K to $475K/year. Work as a Research Intern, Frontier Agents (Summer 2027), Research Intern, Frontier Agents (Winter 2027), and Research Intern, Inference (Summer 2027).

JAX is often requested for Software Engineer jobs roles.

Receive emails of JAX AI Jobs in San Francisco, California, United States
Share:
AI Jobs
JAX Jobs in AI
JAX AI Jobs in San Francisco, California, United States
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
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 9D
TML
Research, Mid Training
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 25D
D
Staff Software Engineer, Machine Learning (Consumer Revenue)
Discord / San Francisco, United States / Remote / Staff
Python jobs PyTorch jobs TensorFlow jobs
$272K – $340K 25D
TA
Research Intern, Frontier Agents (Summer 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs Reinforcement Learning jobs
$58/hr – $70/hr 28D
TA
Research Intern, Frontier Agents (Winter 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs Reinforcement Learning jobs
$58/hr – $70/hr 28D
TA
Research Intern, Inference (Winter 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs CUDA jobs
$58/hr – $70/hr 28D
TA
Research Intern, Inference (Summer 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs CUDA jobs
$58/hr – $70/hr 28D

Frequently asked questions

What is the average salary for JAX jobs?

8 active listings have salaries ranging from $120K to $475K/year.

Which companies are hiring for JAX jobs?

Top companies hiring include Together AI, Discord, and Thinking Machines Lab.

Are there remote JAX jobs?

Yes, 1 of 8 are remote.