Senior Research Engineer

Summary

Join our team as a Senior Research Engineer to contribute to cutting-edge Speech and Natural Language Processing (NLP) research, leveraging expertise in deep learning, JAX, and distributed systems.

Requirements

• 2+ years of professional experience in deep learning research & development
• Hands-on experience with JAX/TPUs and distributed training
• Excellent knowledge of GPU and TPU hardware
• Strong Python programming skills
• Expertise in deep learning architectures and techniques such as CNNs, Transformers, SSMs, self-supervised learning, etc
• Experience with training large Transformer models on distributed systems
• Strong written and verbal communication skills for technical matters

Responsibilities

• Train large-scale Speech AI models, including speech-focused multi-modal LLMs and ASR models with billions of parameters
• Write and optimize training code for maximum efficiency and memory utilization
• Stay up-to-date on the latest AI research and share insights across the company
• Collaborate with the technology leadership team to prioritize the research and engineering agenda, define project scopes, and lead their execution

Benefits

Salary range: $185K - $260K