Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

XLA Jobs with Paid Time Off in AI

Browse 4 XLA jobs with Paid Time Off in AI 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 Jobs with Paid Time Off in AI

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 Product Owner Program Manager SGLang SIEM Slurm TCP/IP XLA 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

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 with Paid Time Off in AI?

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

Which companies are hiring for XLA jobs with Paid Time Off in AI?

Top companies hiring include Thinking Machines Lab.