Research Engineer, NLP Modeling (Remote) at AssemblyAI

The Role:

AssemblyAI is growing quickly, and we’re searching for a Research engineer specializing in NLP Modeling to join our NLP team. With significant investment and strong leadership to fuel our growth, it’s the perfect time to join the AssemblyAI team!

In this role you’ll have the opportunity to:

Replicate state-of-the-art Deep Learning models based on conference/journal publications
Drive best practices for the team in terms of model optimization and maximizing accelerator utilization
Stay up to date on model innovations related to Natural Language Understanding such as Transformer variants (sparse, linear, LSH attention) and new architectures (S4)
Act as a bridge between research and engineering to ensure our models perform well in a production environment

Our Team:

We are a fully remote team made up of problem solvers, innovators and top AI researchers with 20+ years of experience in Machine Learning, Speech Recognition, and NLP from places like DeepMind, Google, Meta, Amazon, Apple, and Cisco. Our culture is super collaborative, low-ego, transparent, and fast-paced. We want to win - and have a flat organization where everyone can openly share ideas (regardless of their title or position) in order to get the best idea.

As a remote company, our team members are given a lot of trust and autonomy to work where and how they want. We look for people to join our team who are ambitious, curious, and self-motivated, and we put a lot of trust and autonomy into everyone on our team. We want to empower everyone to do their best work with whatever tools, structures, or resources they need to perform at their highest potential.

Requirements

3+ years of non-internship professional software development experience in developing and debugging C/C++ or Python
2+ years of experience coding neural network models using primitives (matmul, einsum, relu, softmax, etc.) offered by modern Deep Learning frameworks such as PyTorch, Jax, or TensorFlow
Knowledge of professional software engineering best practices
Excellent written and oral communication skills in English
BSc or MSc Degree in Computer Science, Electrical Engineering, or other technical field

Preferred Qualifications:

2+ years of experience in optimization engineering related to machine learning or computational science
Solid knowledge of data structures and algorithms