# Member of Technical Staff — Inference-TPU at RadixArk

- Company: RadixArk
- What the company does: RadixArk builds large-scale inference and training systems for the entire AI community, making frontier-level AI infrastructure open and accessible. Backed by Accel.
- Company website: https://www.radixark.ai/
- Type: Startups (AI role)
- Level: Senior
- Location: Palo Alto, CA
- Work setup: On-site
- Pay: $200K to $400K base salary per year (USD)
- Posted: 2025-12-29
- Apply by: 2026-10-08
- Apply: https://job-boards.greenhouse.io/radixark/jobs/4087890009
- Page: https://www.1752.vc/careers/jobs/radixark-member-of-technical-staff-inference-tpu/

## About the role

RadixArk is looking for a Member of Technical Staff — TPU Systems to build high-performance inference and training systems using JAX, XLA, and Pallas. You'll push model workloads to their limits on TPU hardware, working on SGLang-JAX and other critical infrastructure that enables efficient deployment of frontier models on Google's tensor processing units.

## What they're looking for

- 3+ years experience building production ML systems utilizing JAX/Torch, XLA, or TPU-focused frameworks
- Bachelor's or Master's degree in Computer Science, Electrical Engineering, or equivalent industry experience
- Deep understanding of XLA internals preferred: HLO, MLIR, operator fusion, SPMD partitioning, and sharding strategies
- Strong performance tuning instincts across compiler and runtime layers
- Experience with distributed inference systems (e.g. SGLang, vLLM) or training frameworks (e.g. Miles, Alpa, Pathways)
- Proficiency in Python with demonstrated ability to write high-performance, production-quality code

Tags: Member of Technical Staff