Sign Up Sign In

Post a Job

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

AI Engineer Ansible Customer Success Manager Cypress Data Engineer Data Scientist Elasticsearch Engineering Manager Ethernet HR JAX MongoDB NCCL OpenTelemetry Platform Engineer Product Manager Program Manager RDMA React Rust SEO Snowflake Tableau TensorRT-LLM Triton TypeScript

Cities

San Francisco Seattle 3 Santa Clara 2 Amsterdam 2 Toronto 2 Cambridge 2

Countries

United States 18 California 12 Washington 3 Netherlands 2 Canada 2 United Kingdom 2

Filter by type

Remote Full Time Internship

C

Research Engineer, Safety & Alignment

Cognition / San Francisco, United States

Python PyTorch JAX

8D

TML

RL Systems - Infra

Thinking Machines Lab / San Francisco, United States

PyTorch JAX RLHF

$425K – $475K 9D

TML

Research, Mid Training

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch TensorFlow JAX

$350K – $475K 25D

D

Staff Software Engineer, Machine Learning (Consumer Revenue)

Discord / San Francisco, United States / Remote / Staff

Python PyTorch TensorFlow

$272K – $340K 25D

TA

Research Intern, Frontier Agents (Summer 2027)

Together AI / San Francisco, United States / Entry

PyTorch JAX Reinforcement Learning

$58/hr – $70/hr 29D

TA

Research Intern, Frontier Agents (Winter 2027)

Together AI / San Francisco, United States / Entry

PyTorch JAX Reinforcement Learning

$58/hr – $70/hr 29D

TA

Research Intern, Inference (Winter 2027)

Together AI / San Francisco, United States / Entry

PyTorch JAX CUDA

$58/hr – $70/hr 29D

TA

Research Intern, Inference (Summer 2027)

Together AI / San Francisco, United States / Entry

PyTorch JAX CUDA

$58/hr – $70/hr 29D

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.