AI Jobs
AI Jobs
Post a job
Sign up Sign in

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

Browse 12 AI jobs in San Francisco for Aug 2026 with roles open at Thinking Machines Lab paying $350K to $475K/year. Work as a Research Engineer, Infrastructure, Numerics, Research Engineer, Infrastructure, Inference, and Research, Audio Expertise.

JAX is often requested for Research Engineer roles.

Receive emails of Full Time JAX AI Jobs in San Francisco, California, United States
Share:
AI Jobs
JAX Jobs in AI
Full Time 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 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 Sketch jobs Tableau jobs TensorRT-LLM jobs Triton jobs TypeScript jobs
Filter by type
Full Time
TML
Research, Post-Training
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Inference
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs vLLM jobs
$350K – $475K 13D
TML
Research, Pre-Training Data
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research, Pre-Training Science
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research, Audio Expertise
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Numerics
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs Distributed Systems jobs
$350K – $475K 13D
TML
Research, Vision Expertise
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, RL Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs RLHF jobs
$350K – $475K 13D
TML
Software Engineer, Supercomputing
Thinking Machines Lab / San Francisco, United States / Hybrid
Python jobs PyTorch jobs TensorFlow jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs CUDA jobs
$350K – $475K 13D
TML
Research, Post-Training Data
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Training Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs Megatron jobs
$350K – $475K 13D

Frequently asked questions

What is the average salary for Full Time JAX jobs?

12 active listings have salaries ranging from $350K to $475K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include Thinking Machines Lab.