Senior Researcher - Deep Learning

About the role:

We're looking for a Senior Researcher to join our Research team. Creating best-in-class AI models and deploying them in production constitute the core competencies of AssemblyAI. This person will conduct applied research in Speech and Natural Language Processing (NLP) in a team setting. Specific responsibilities include: learning new ideas from research and technology communities, writing code to implement new ideas, training new models, meticulously evaluating these models, and collaborating with engineers to deploy models in production. This person should have a strong background in (applied) deep learning research and should be able to write high-quality code. With a deep understanding of the fundamentals of deep learning and its applications, they will contribute to other projects through discussions as well. 

What You’ll Do:

• Provide scientific leadership on ambitious research projects that have the potential for high impact on AssemblyAI's products and the state of applied AI technology

• Engage with research communities through activities such as publishing papers or participating in conferences

• Execute those research projects in a scientifically sound manner

• Stay up-to-date on the latest AI research and share this knowledge across the company

• Collaborate with the technology leadership team to prioritize the research agenda, define the scope of research projects, and take charge of their execution

• Collaborate with the engineering team to deploy new models into production

What You’ll Need:

• Master's degree in Computer Science or a related field

• 2+ years of professional experience in Deep Learning research & development

• Excellent knowledge of deep learning architectures and techniques such as CNNs, Transformers, SSMs, self-supervised learning, etc. 

• 2+ years of hands-on experience with PyTorch or JAX

• Strong written and verbal communication skills in technical matters

Nice to Have:

• Experience with JAX/Pallas/TPUs

• Experience in training large (Transformer) models on distributed systems

• Experience in training/fine-tuning (multimodal) Large Language Models

• Expertise in writing high performance CPU/GPU/TPU code

• Track record of publications in deep learning or its applications at top-tier conferences

You May Be a Great Fit if:

• Highly self-motivated and keen to have real-world impact

• Detail-oriented, analytical, and creative problem solver

• Meticulous in all aspects (code, data, modeling, ...)

• Able to work effectively in a team-oriented, collaborative environment

Pay Transparency:

AssemblyAI strives to recruit and retain exceptional talent from diverse backgrounds while ensuring pay equity for our team. Our salary ranges are based on paying competitively for our size, stage and industry, and are one part of many compensation, benefits and other reward opportunities we provide.

There are many factors that go into salary determinations, including relevant experience, skill level and qualifications assessed during the interview process, and maintaining internal equity with peers on the team. The range shared below is a general expectation for the function as posted, but we are also open to considering candidates who may be more experienced than outlined in the job description. In this case, we will communicate any updates in the expected salary range.

Lastly, the provided range is the expected salary for candidates in the U.S. Outside of those regions, there may be a change in the range, which again, will be communicated to candidates.

Salary range: $190K - $247K