A
$270,000 – $310,000
Listed on AssemblyAI’s own careers site. You apply with them directly — we never stand between you and the employer.
What this role is
A Senior Research Engineer role focused on optimizing large-scale distributed training, data processing, and inference systems for a voice AI platform. This suits engineers with deep expertise in JAX, TPUs, and ML infrastructure who want to directly impact both research velocity and production systems at a capital-efficient, fast-moving company.
Our summary, not AssemblyAI’s wording. The full posting is on their site.
Skills this role names
Log in to see which of these are already on your profile.
What they ask for
Required
- Expert-level proficiency with JAX and TPUs
- Measurement discipline and skepticism of results
- Experience optimizing inference systems for production
- Deep understanding of distributed training and modern deep learning systems
- Familiarity with inference optimization techniques like quantization and KV-cache management
- Strong Python skills
- Excellent communication and collaborative mindset
Nice to have
- Domain knowledge in Speech-to-Text and ASR architectures
- Audio processing expertise
- C++ or Rust experience for kernel-level work