Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

XLA AI Jobs in United States

Browse 5 AI jobs in United States for Aug 2026 with roles open at Cohere and Thinking Machines Lab paying $350K to $475K/year. Work as a Member of Technical Staff, Agentic Environments , Research Engineer, Infrastructure, Numerics, and Research Engineer, Infrastructure, Inference.

XLA is often requested for Research Engineer roles.

Receive emails of XLA AI Jobs in United States

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

Cities

San Francisco 4 New York 1

Countries

United States California 4 Canada 1 New York 1

Filter by type

Remote Full Time

C

Member of Technical Staff, Agentic Environments

Cohere / Canada ; New York, NY / Remote / Staff

Python PyTorch JAX

3D

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?

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

Which companies are hiring for XLA jobs?

Top companies hiring include Cohere and Thinking Machines Lab.

Are there remote XLA jobs?

Yes, 1 of 5 are remote.