Software Engineer, Research

AssemblyAI is seeking a mid-level software engineer to join their team and help create and manage their Deep Learning research framework. The ideal candidate should have strong software and cloud engineering skills, as well as experience in building maintainable systems. Collaboration skills are also important, as the engineer will work closely with the Research team and lead a small team for larger projects. The responsibilities include designing and integrating the experiment framework, maintaining databases and documentation, and ensuring the platform is well-tested and resilient. This role offers the opportunity to work on a system that enables large-scale deep learning research and be part of a small, cross-functional team.

• Help design the new experiment framework and integrate it with an open source management platform
• Enable researchers to launch many experiments in the cloud across 100s of accelerators by running a single shell script
• Design, implement, and maintain the experiment framework, databases, and documentation for researchers
• Ensure that model code is hermetically packaged for easy deployment to production
• Ensure that the platform is well tested and resilient to failures and capacity issues

• 3+ years of engineering backend applications using Python and/or other backend language(s) such as Java, C#, JavaScript, Go, C/C++
• 2+ years of working with SQL and NoSQL databases
• 2+ years working with common AWS or GCP services, or a similar platform
• 2+ years of being a maintainer of a heavily used library or framework
• 2+ years of working with accelerator backed compute (GPU or TPU)
• Experience with bazel as a build system

• Competitive salary range of $140,000-$170,000 USD
• Remote work opportunity
• Opportunity to work with a leading Applied AI company
• Chance to work on cutting-edge AI models for audio data transcription and understanding
• Exposure to the latest developments and breakthroughs in AI research
• Opportunity to collaborate with a world-class, remote team
• Chance to contribute to the growth of an emerging market leader in applied AI
• Ownership of a mission-critical software
• Opportunity to work on a system that enables large-scale deep learning research
• Small, cross-functional team environment
• Equal opportunity and commitment to diversity and inclusion