Senior Research Engineer - Large-Scale DL Systems (JAX/TPU)

AssemblyAI is seeking a Senior Research Engineer to advance our large-scale distributed training, data processing, and inference systems. You will move quickly from experiment to measurement to deployment, raising the ceiling on what is possible and helping shape the core platform that powers our API.

The role demands deep learning systems expertise and strong engineering skill in JAX and TPUs, with a focus on low-latency, asynchronous inference, layer-level optimization, and scalable