xla

Recent positions

• Member of Technical Staff - Multimodal Understanding

  xai · Apr 17, 2026

  $180k – $440k

  senior · on-site · Palo Alto

  c++ python pytorch kubernetes jax rust +3
  Apply

• Member of Technical Staff - Compilers

  gimlet · Mar 10, 2026

  $180k – $400k

  senior · on-site · full time · San Francisco

  c++ python mlir tvm llvm triton +3
  Apply

• Staff Compiler Engineer

  olix · Feb 10, 2026

  $225k

  senior · on-site · full time · Toronto

  c++ pytorch tensorflow jax mlir tvm +4
  Apply

• Member of Technical Staff, ML Kernels

  netpreme · Feb 6, 2026

  senior · on-site · full time · Boston, Santa Clara

  c++ pytorch cuda tensorflow xla nvprof +1
  Apply

• Staff Compiler Engineer

  olix · Feb 1, 2026

  £306k

  staff · on-site · full time · London

  c++ pytorch tensorflow jax mlir tvm +4
  Apply

• Member of Technical Staff, AI Research

  psi · Jan 26, 2026

  senior · on-site · full time · Boston

  pytorch jax ray accelerate xla sac +3
  Apply

• Member of Technical Staff, Kernel Engineering

  inferact · Jan 22, 2026

  $200k – $400k

  senior · on-site · full time · San Francisco

  c++ python cuda nvidia mlir intel +5
  Apply

• Senior Engineering Leader - AI Infrastructure and Inferencing

  gruve · Jan 15, 2026

  $240k – $300k

  director · on-site · full time · Redwood City

  c++ cuda aws docker kubernetes gcp +9
  Apply

• Research Engineer, Interpretability

  anthropic · Nov 7, 2025

  $315k – $560k

  senior · on-site · full time · San Francisco

  python pytorch cuda jax gpu transformers +2
  Apply

• Senior Software Engineer, Post-Training & RL Frameworks

  waymo · Oct 22, 2025

  $204k – $259k

  senior · hybrid · full time · Mountain View

  c++ python tensorflow jax xla
  Apply

• SDE IV - GPU Engineer

  glance · Oct 9, 2025

  $150k – $200k

  senior · on-site · full time · Bangalore

  c++ cuda mlir tvm triton xla +1
  Apply

• Member of Technical Staff, MLE (UK/EU)

  cohere · Sep 30, 2025

  £90k – £120k

  senior · remote · full time · London

  python tensorflow jax mlir xla tf-serving
  Apply

• Performance Engineer, GPU

  anthropic · Sep 22, 2025

  $280k – $850k

  senior · hybrid · full time · San Francisco

  pytorch cuda jax nsight triton xla +4
  Apply

• Senior Software Engineer, ML Evaluation Infra and Efficiency

  waymo · Aug 26, 2025

  $238k – $302k

  senior · on-site · full time · Mountain View

  c++ python tensorflow jax gpu triton +4
  Apply

• Member of Technical Staff, Integration/RL Team (Research Engineer)

  cohere · Aug 19, 2025

  $1500k – $2000k

  senior · remote · full time · Paris

  python pytorch kubernetes jax mlir ray +1
  Apply

• AI Runtime Engineer

  enchargeai36 · Jul 10, 2025

  mid · remote · full time · Remote

  c++ tensorrt vllm mlir tvm openvino +4
  Apply

• Member of Technical Staff, Post-Training

  cohere · Jun 13, 2025

  senior · remote · full time · London

  python pytorch kubernetes jax mlir ray +2
  Apply

• Member of Technical Staff, Search

  cohere · May 23, 2025

  senior · remote · full time · San Francisco

  python pytorch tensorflow jax mlir xla
  Apply

• ML Runtime Optimization Engineer

  appliedintuition · Mar 5, 2025

  $159k – $199k

  mid · on-site · full time · Sunnyvale

  pytorch tensorrt cuda jax gpu soc +4
  Apply

• Member of Technical Staff, Training Performance Engineer

  cohere · Feb 20, 2025

  £90k – £120k

  mid · remote · full time · London

  python pytorch cuda jax mlir triton +1
  Apply