Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

JAX AI Jobs in San Jose, California, United States

Browse 2 AI jobs in San Jose for Aug 2026 with roles open at Figure and SambaNova. Work as a Helix AI Engineer, Training Performance and Principal AI Systems Performance Engineer.

JAX is often requested for Research Engineer roles.

Receive emails of JAX AI Jobs in San Jose, California, United States

AI Engineer Bash CUDA Compiler Engineer Data Engineer Ethernet HR JAX MLflow NCCL OpenCV Program Manager Prototyping Python RDMA React Reinforcement Learning Rust SEO Sketch Spark Tableau Technical Documentation TensorRT-LLM Triton TypeScript

Cities

San Francisco 12 San Jose Palo Alto 1 Cambridge 1 Glen Burnie 1 New York 1

Countries

United States 18 California 15 Massachusetts 1 Maryland 1 Canada 1 New York 1

Filter by type

Full Time

S

Principal AI Systems Performance Engineer

SambaNova / San Jose, United States / Staff

PyTorch TensorFlow JAX

23h

F

Helix AI Engineer, Training Performance

Figure / San Jose, United States

Python PyTorch JAX

$200K – $400K 3D

Frequently asked questions

Which companies are hiring for JAX jobs?

Top companies hiring include Figure and SambaNova.