Senior Researcher, Deep Learning

Summary

Join AssemblyAI's Research team as a Senior Researcher and contribute to the development of cutting-edge Speech AI models. You will be responsible for training large-scale models, optimizing code, evaluating model performance, and collaborating with engineers. This role requires expertise in deep learning research and engineering, including experience with JAX/TPUs and distributed training. The ideal candidate will possess strong Python programming skills and excellent communication abilities. AssemblyAI offers a competitive salary and benefits package, and is a fully remote team.

Requirements

• 2+ years of professional experience in deep learning research & development
• Demonstrated experience in all aspects of deep learning model development, including data acquisition and processing, implementation, model training, experimental analysis, and writing inference and evaluation code as an individual contributor
• Hands-on experience with JAX/TPUs and distributed training
• Excellent knowledge of GPU and TPU hardware
• Strong Python programming skills
• Strong written and verbal communication skills for technical matters

Responsibilities

• Train large-scale Speech AI models, including ASR and speech-focused multi-modal LLMs 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

Preferred Qualifications

• Experience in training/fine-tuning (multimodal) Large Language Models
• Experience in LLM inference using frameworks like vLLM or JetStream

Benefits

• Salary range: $215K - $275K
• Fully remote team