Member of Technical Staff — TPU Systems (JAX / XLA / PALLAS)

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. Requirements 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 D