Senior Researcher, Deep Learning

February 16

Apply Now

Receive Emails with Similar Jobs

AssemblyAI

WebsiteLinkedInAll Job Openings

AssemblyAI is a Speech AI company focused on building new state-of-the-art AI models that can transcribe and understand human speech. Our customers, such as CallRail, Fireflies, and Spotify, choose AssemblyAI to build incredible new AI-powered experiences and products based on voice data.

51 - 200 employees

📋 Description

• At AssemblyAI, we’re building at the forefront of Speech AI, creating powerful models for speech-to-text and speech understanding available through a straightforward API. • Progress in AI is moving at an unprecedented pace– and our team is made up of experts in AI research that are focused on making sure that our customers are able to stay on the cutting edge, with production-ready AI models that are constantly updating and improving as our team continues to improve accuracy, latency, and what’s possible with Speech AI. • We are seeking a Senior Researcher to join our Research team. AssemblyAI's core strengths include developing best-in-class AI models in Speech and Natural Language Processing (NLP), writing optimized inference code, and serving models at scale with low latency and high availability. • This role demands expertise in both applied research and engineering in Deep Learning. Key responsibilities include: Writing efficient, high-performance training code for large-scale models with billions of parameters on TPU clusters, optimizing code for both training and inference, rigorously evaluating models to ensure quality and performance, and collaborating closely with engineers to support data creation, data filtering, and model deployment. • The ideal candidate will possess a fine-grained understanding of deep learning research and engineering, spanning both software and hardware.

🎯 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. • Experience in training/fine-tuning (multimodal) Large Language Models. • Experience in LLM inference using frameworks like vLLM or JetStream.