XLA jobs
Open developer roles that use XLA, pulled straight from company job boards and refreshed every few hours.
19
open roles
19
posted this week
6
fully remote
10
pay $150k+
Latest openings
Filter in search →
Model Engineer, Multimodal Understanding
SpaceXAI - Palo Alto, CA - On-site
$160k–250k3 days ago
Senior Staff/Principal Software Engineer - GPU System Software, Onboard Execution
Zoox - Foster City, CA +2 - On-site
$339k–455k3 days ago
Engineering Manager, Performance Optimization
Zoox - Foster City, CA +1 - On-site
$237k–322k3 days ago
Engineering Manager, ML Performance Optimization
Zoox - Foster City, CA - On-site
$276k–343k3 days ago
Member of Technical Staff, AI-Driven Compilation
San Francisco Tensor Company - San Francisco, CA - On-site
$275k–315k3 days ago
Member of Technical Staff, GPU Compiler
San Francisco Tensor Company - San Francisco, CA - On-site
$285k–315k3 days ago
Senior / Staff AI Research Engineer, Real-Time Inference
RoboForce - Milpitas, CA - On-site
$185k–325k3 days ago
Member of Technical Staff - Compiler Engineer
Gimlet Labs - San Francisco, CA - On-site
$180k–400k3 days ago
Multimodal AI Model Optimization Research Engineer
Tavus - San Francisco, CA - Remote
3 days ago
Member of Technical Staff, TPU Kernel Engineering
Inferact - San Francisco, CA - On-site
$200k–400k3 days ago
Applied Researcher – AI Expert
Designworks Talent - Bellevue, WA - Hybrid
4 days ago
Compiler Frontend Engineer
Normal Computing Corporation - Mountain View, CA +1 - Hybrid
4 days ago
Compiler Backend Engineer
Normal Computing Corporation - Mountain View, CA +1 - Hybrid
$200k–300k4 days ago
Senior Inference Optimization ML Engineer
Rhoda AI - Mountain View, CA - On-site
4 days ago
Member of Technical Staff, Integration/RL Team (Research Engineer)
Cohere - New York, NY +2 - Remote
4 days ago
Member of Technical Staff, Post-Training
Cohere - San Francisco, CA +3 - Remote
4 days ago
Member of Technical Staff, Training Infra Engineer
Cohere - San Francisco, CA +3 - Remote
4 days ago
Member of Technical Staff, Training Performance Engineer
Cohere - New York, NY +2 - Remote
4 days ago
Member of Technical Staff, Modeling
Cohere - San Francisco, CA +3 - Remote
4 days ago