[Remote] Senior Research Engineer at AssemblyAI

Candidates must have strong expertise in the Python ecosystem and major ML frameworks like PyTorch and JAX, along with experience in lower-level programming languages such as C++ or Rust. A deep understanding of GPU acceleration, including CUDA, profiling, and kernel-level optimization, is essential, with TPU experience being a strong plus. Proven ability to accelerate deep learning workloads using compiler frameworks, graph optimizations, and parallelization strategies is required, as is a solid understanding of the deep learning lifecycle from model design to inference deployment. Strong debugging, profiling, and optimization skills in large-scale distributed environments are necessary, alongside excellent communication and collaboration skills.