Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Full Time JAX AI Jobs in San Francisco, California, United States

Browse 4 AI jobs in San Francisco for Oct 2026 with roles open at Discord, Thinking Machines Lab, and Cognition paying $272K to $475K/year. Work as a Staff Software Engineer, Machine Learning (Consumer Revenue), Research, Mid Training, and RL Systems - Infra.

JAX is often requested for Software Engineer jobs roles.

Receive emails of Full Time 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 Cambridge 2 Cambridge 1 San Jose 1 Washington 1

Countries

United States 12 California 6 Washington 3 United Kingdom 2 Massachusetts 1 District of Columbia 1

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

Frequently asked questions

What is the average salary for Full Time JAX jobs?

4 active listings have salaries ranging from $272K to $475K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include Discord, Thinking Machines Lab, and Cognition.

Are there remote Full Time JAX jobs?

Yes, 1 of 4 are remote.