Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

XLA Jobs in AI

Browse 5 XLA jobs in AI 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 Jobs in AI

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

Filter by benefits

🦷 Dental Insurance 🤓 Vision Insurance 💰 401k 🏖 Unlimited Vacation 🏖 Paid Time Off 🏔 Company Retreats 🖥 Home Office Budget 📚 Learning Budget 🧘 Mental Wellness Budget ✈️ Visa Sponsorship 📦 Relocation

Cities

San Francisco 4 New York 1

Countries

United States 5 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 in AI?

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

Which companies are hiring for XLA jobs in AI?

Top companies hiring include Cohere and Thinking Machines Lab.

Are there remote XLA jobs in AI?

Yes, 1 of 5 are remote.