About the job
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
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