Join AssemblyAI's Research team to develop and improve systems behind large-scale distributed training, data processing, and inference, working with JAX and TPUs. The role is cross-functional, embedded in the Research team, raising experimental velocity across the ML pipeline.