Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Full Time JAX AI Jobs in Palo Alto, California, United States

Browse 3 AI jobs in Palo Alto for Nov 2026 with roles open at xAI paying $180K to $600K/year. Work as a Sr. Model Engineer, Mid-training, Sr. Model Engineer, Post-Training & RL, and Sr. Model Engineer, RL Training Framework.

JAX is often requested for ML Engineer jobs roles.

Receive emails of Full Time JAX AI Jobs in Palo Alto, California, United States

Android Engineer Bash CUDA Data Engineer Datadog Ethernet Head of Design JAX Kotlin Kubernetes NCCL Product Manager Python RDMA RLHF Ray React Reinforcement Learning Rust Snowflake Spark TensorRT-LLM Triton TypeScript User Research vLLM

Cities

San Francisco 6 Bristol 4 New York 4 San Jose 3 Palo Alto London 3 Austin 2

Countries

United States 18 California 14 United Kingdom 6 New York 4 Texas 2

Filter by type

Full Time

X

Sr. Model Engineer, RL Training Framework

xAI / Palo Alto, United States / Senior

Python PyTorch JAX

$180K – $400K 20h

X

Sr. Model Engineer, Post-Training & RL

xAI / Palo Alto, United States / Senior

Python PyTorch JAX

$180K – $600K 3D

X

Sr. Model Engineer, Mid-training

xAI / Palo Alto, United States / Senior

PyTorch JAX Docker

$180K – $440K 3D

Frequently asked questions

What is the average salary for Full Time JAX jobs?

3 active listings have salaries ranging from $180K to $600K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include xAI.