"xla" Jobs

3 open tech roles matching “xla”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: Python, CUDA, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 3 of 3 results

Senior Research Engineer

Join AssemblyAI as a Senior Research Engineer to enhance large-scale distributed training and inference systems in Voice AI.

AssemblyAI Remote - New York $270k–$310k/yr Published 1 week ago

C++ JAX Rust TPU Python

Flexible on stack

Apply →

Performance Engineer, GPU Hybrid Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 11 months ago

FP8 NVLink Triton JAX CUTLASS

Apply →

Staff+ Software Engineer, Inference Velocity Hybrid Visa

Join Anthropic as a Staff Engineer to lead the technical direction of the Inference Runtime for AI systems serving millions of users.

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | Seattle, WA | New York City, NY $405k–$485k/yr Published 3 months ago

CUDA Trainium Rust AWS Neuron TPU

Flexible on stack

Apply →