Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

XLA AI Jobs in California

Browse 4 AI jobs in California 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 Engineer, Infrastructure, Training Systems.

XLA is often requested for Research Engineer roles.

Receive emails of XLA AI Jobs in California

AI Engineer Customer Success Manager Data Engineer Data Scientist DeepSpeed Engineering Manager Excel Forward Deployed Engineer Full Stack Engineer GCP HR InfiniBand JavaScript Looker MySQL PagerDuty Photoshop Product Manager Program Manager SGLang SIEM Slurm TCP/IP Terraform XLA Zendesk

Cities

San Francisco 4 New York 1

Countries

United States 5 California Canada 1 New York 1

Filter by type

Full Time

TML

Research Engineer, Infrastructure, Inference

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX vLLM

$350K – $475K 12D

TML

Research Engineer, Infrastructure, Numerics

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX Distributed Systems

$350K – $475K 12D

TML

Research Engineer, Infrastructure, Kernels

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX CUDA

$350K – $475K 12D

TML

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX Megatron

$350K – $475K 12D

Frequently asked questions

What is the average salary for XLA jobs?

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

Which companies are hiring for XLA jobs?

Top companies hiring include Thinking Machines Lab.