XLA Jobs

SF Tensor is revolutionizing high-performance compute for AI by rebuilding the entire stack, from hardware to cloud. We are developing a Kernel Optimizer that automatically finds the fastest code for any vendor or cluster topology, and a Model Foundry to manage runs, simplify research, and enable workload portability across clouds and chips. Our mission is to eliminate the compute tax, making AI development faster, cheaper, and more accessible. We are backed by prominent investors and industry leaders, and we seek individuals who believe that advancements in AI are contingent on leaps in compute infrastructure.

$275k - $315k

Anthropic

Anthropic's Inference organization is responsible for delivering Claude to millions of users and enterprise customers with the speed, reliability, and efficiency that frontier AI demands. We are seeking a Staff engineer to lead the Inference Developer Productivity team. This team is dedicated to making every engineer in the organization significantly more effective at building, testing, and shipping inference software. This senior individual contributor role involves broad technical ownership, where you will set the technical direction for the team's toolchains, workflows, and feedback loops. You will be responsible for making critical decisions on architecture, prioritization, and trade-offs across heterogeneous accelerator platforms. You will partner closely with the Engineering Manager, who handles hiring and people development, while you drive the technical roadmap and execution. Collaboration with Anthropic's central Infrastructure organization is also key to ensure Inference's multi-accelerator reality is well-supported without duplicating efforts.

$405k - $485k

Thinking Machines is seeking an infrastructure research engineer to design, optimize, and scale the systems that power large AI models. The goal is to make inference faster, more cost-effective, more reliable, and more reproducible, enabling teams to focus on advancing model capabilities. This role is crucial for ensuring that every experiment, evaluation, and deployment runs smoothly at scale, with a focus on performant and efficient model inference for both real-world applications and research acceleration.

$350k - $475k

Thinking Machines is seeking an infrastructure research engineer to design and build the core systems that enable scalable, efficient training of large models for deployment and research. The goal is to make experimentation and training fast and reliable, allowing research teams to focus on science rather than system bottlenecks. This role is ideal for individuals with deep systems and performance expertise combined with a curiosity for machine learning at scale, who will own the training stack end-to-end to ensure every GPU cycle drives scientific progress.

$350k - $475k

Applied Intuition is seeking a software engineer with deep expertise in optimizing ML models and deploying them within production-grade embedded runtime environments. This role involves working across the entire ML framework stack, including PyTorch, JAX, ONNX, TensorRT, CUDA, XLA, and Triton. You will be instrumental in driving ML performance optimization for various technologies used in ADAS/AD stacks, targeting diverse embedded compute platforms. The position requires developing strategies to enhance the efficiency and reduce the latency of model inference on customer-selected compute boards, and supporting deployment on memory-constrained platforms through model pruning and quantization.

$217k - $318k

Cohere

Cohere is a leading enterprise AI company building cutting-edge foundation AI models and end-to-end products. We are seeking an engineer to provide strong support for model training pipelines, ship state-of-the-art models to production, and bridge the gap between research and production. This role offers access to significant compute resources, data, and talent, with a collaborative environment where engineers contribute to both production code and research efforts based on interest and organizational needs. We are a global company with a remote-friendly approach, valuing exceptional talent regardless of location.

CA$250k - CA$535k