TECHNOLOGY · AI & MACHINE LEARNING

XLA

Domain-specific compiler for linear algebra developed by Google that optimizes and accelerates machine learning computations in TensorFlow and JAX.

Open roles mentioning it
41
Companies hiring for it
16
Remote share
5%
WHO'S HIRING

Companies with open roles mentioning XLA

Cohere9 open rolesWaymo5 open rolesRadixArk4 open rolesAnthropic3 open rolesMatX3 open rolesRiver AI2 open rolesSF Tensor2 open rolesVelaura AI2 open rolesFireworks AI2 open rolesHudson River Trading2 open rolesZoox2 open rolesHiggsfield1 open role
APPEARS ALONGSIDE

Technologies mentioned in the same job posts

PyTorch×27CUDA×27JAX×26Python×24Triton×20C++×18TensorFlow×10Docker/Kubernetes×9LLVM×9Transformers×8
COMMON ROLES

Job titles that ask for XLA

  • Member Of Technical Staff, Performance Optimization2
  • Runtime Engineer2
  • Ai Research Engineer, Pre-Training1
  • Founding Gpu Compiler Engineer1
  • Founding Research Engineer, Ai-Driven Compilation1
  • Machine Learning Engineer, Runtime & Optimization1
  • Machine Learning Intern/Co-Op (Winter 2027)1
  • Applied Machine Learning Engineer1
GO DEEPER

Search these companies

Every technology on this page is extracted from live job descriptions — not self-reported marketing. Filter companies by tech stack or create an account to track the ones hiring for yours.