[HOME/SKILLS/JAX/SAN FRANCISCO BAY AREA]

LIVE FEEDUTC 2026.08.20

// SKILL · JAX · SAN FRANCISCO BAY AREA

JAX jobs in San Francisco Bay Area[9 OPEN]

Every open role in San Francisco Bay Area that lists JAX in its stack, pulled directly from each employer.

[ OPEN ROLES ]

9

[ NEW · 30D ]

3

[ TOP COMPANIES ]

5

§ 01

OPEN ROLES

SORTED BY POSTED · TOP 9

Senior Reinforcement Learning Engineer

Apptronik·SUNNYVALE, CA·ON-SITE

[MUJOCO][PYTORCH][JAX][REINFORCEMENT LEARNING]

$230,000 – $260,000

Staff/Lead Machine Learning Engineer, Behavior & Planning

Nuro·MOUNTAIN VIEW, CALIFORNIA (HQ)·ON-SITE

[PYTORCH][TENSORFLOW][JAX][IMITATION LEARNING]

Machine Learning Engineer, Perception

Wing·PALO ALTO, CALIFORNIA·ON-SITE

[COMPUTER VISION][PYTORCH][TENSORFLOW][JAX]

$190,000 – $202,000

Senior Perception Learning Engineer - SLAM

Apptronik·SUNNYVALE, C A·ON-SITE

[ROS2][ROS][SLAM][VIO]

$190,000 – $235,000

Research Scientist, Wayve Labs

Wayve·SUNNYVALE, CALIFORNIA USA·ON-SITE

[SLAM][COMPUTER VISION][PYTORCH][JAX]

$230,000 – $380,000

EXECUTIVE3MO AGO

Distillation Lead

Waabi·SAN FRANCISCO, CA·ON-SITE

[COMPUTER VISION][PYTORCH][JAX][CUDA]

$195,000 – $286,000

Behavior Planning Software Engineer

Aurora Innovation·SAN FRANCISCO, CALIFORNIA·ON-SITE

[MOTION PLANNING][PYTORCH][TENSORFLOW][JAX]

$126,000 – $201,000

Senior Perception Learning Engineer

Apptronik·SUNNYVALE, CA·ON-SITE

[ROS2][ROS][SLAM][VIO]

$190,000 – $235,000

ML Infra Engineer

Physical Intelligence·SAN FRANCISCO·ON-SITE

[PYTORCH][JAX][KUBERNETES][AWS]

§ 02

TOP COMPANIES HIRING JAX IN SAN FRANCISCO BAY AREA

TOP 5

§ 03

JAX ROLES IN OTHER CITIES

§ 04

OTHER SKILLS HIRING IN SAN FRANCISCO BAY AREA

[PYTHON · 170][EMBEDDED · 43][LIDAR · 39]