Senior Machine Learning Software Engineer, AssemblyAI

$195-225k

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 will be communicated to candidates throughout the interview process

AWS

Kubernetes

Python

PyTorch

Bazel

Senior and Expert level

Remote from Canada, Europe, UK, US

More information about location

Speec AI models to automatically recognize and understand speech

Open for applications

Speec AI models to automatically recognize and understand speech

101-200 employees

B2BArtificial IntelligenceDeep TechSaaS

Open for applications

$195-225k

AWS

Kubernetes

Python

PyTorch

Bazel

Senior and Expert level

Remote from Canada, Europe, UK, US

More information about location

101-200 employees

B2BArtificial IntelligenceDeep TechSaaS

Company mission

To create new, superhuman Speech AI models that will unlock entirely new classes of applications and products to be built leveraging voice data.

Job

Company

Role

Who you are

This role requires a strong backend engineering background in distributed systems and containerization, and a track record of independently driving projects from concept to delivery
Strong backend engineering experience with Python
Experience building and operating distributed, containerized applications, preferably on AWS
Proficiency implementing observability solutions (monitoring, logging, alerting, tracing) for production systems
Ability to design and implement resilient, scalable architectures
Track record of independently scoping and delivering complex technical projects from problem identification through production deployment
Comfort navigating ambiguity and making pragmatic technical decisions when requirements are unclear or evolving

Desirable

MLOps experience, including familiarity with PyTorch and Kubernetes
Experience working in fast-paced environments where you owned technical direction for an area and drove projects with minimal oversight
Experience collaborating with remote, globally distributed teams
Comfort working across the entire ML lifecycle from model serving to API development
Experience in audio-related domains (ASR, TTS, or other domains involving audio processing)
Experience with other cloud providers
Familiarity with Bazel and monorepos
Experience with alternative ML inference frameworks beyond PyTorch
Experience with other programming languages
Experience mentoring junior engineers or onboarding teammates onto complex systems

What the job involves

We’re looking for a Senior Machine Learning Engineer to accelerate our AI research-to-production pipeline
You’ll build and improve the infrastructure that enables our research team to rapidly deploy and safely test new models, while helping ensure our production inference systems remain efficient, scalable, and reliable
You’ll identify gaps and opportunities in our ML infrastructure, scope solutions to ambiguous technical problems, and help set the technical direction for how we bridge research innovation and production reliability
This is a cross-functional role that requires close collaboration with both research teams developing models and engineering teams supporting the broader platform
Design and implement tooling that enables researchers to quickly deploy and evaluate new models in production
Design, build, and maintain high-performance, cost-efficient inference pipelines, making architectural decisions about scaling, reliability, and cost trade-offs
Proactively identify and resolve infrastructure bottlenecks, proposing and scoping improvements to iteration speed and production reliability
Develop and maintain user-facing APIs that interact with our ML systems
Implement comprehensive observability solutions to monitor model performance and system health
Troubleshoot and lead resolution of complex production issues across distributed systems, driving root-cause analysis and implementing preventive measures
Set the direction for and continuously improve our MLOps practices, identifying the highest-impact opportunities to reduce friction between research and production
Collaborate closely with research and engineering teams to align on technical direction, and help onboard and mentor engineers on ML infrastructure best practices

Our take

Audio has become central to many products on the internet, from the explosive uptake of Zoom to the inclusion of audio and video messaging features on dating apps. In this vein, AssemblyAI has developed AI models for transcribing, understanding and analyzing audio and video data.

Between the summers of 2021 to 2022, AssemblyAI saw the number of developers building with its APIs grow by over 1000%, with customers ranging from start-ups like Veed to global enterprises including NBC Universal and Spotify. Coupling this with a strong in-house pedigree - from the likes of DeepMind, Google Brain and Meta AI - AssemblyAI has built the foundations for a very exciting future.

And with funding under its belt, the company is scaling up its efforts; accelerating its product roadmap whilst building an even better AI infrastructure to speed up AI research and inference engines.

Kirsty

Company Specialist at Welcome to the Jungle

Insights

Top investors

Glassdoor (4.6)

Company

Funding (last 2 of 5 rounds)

Dec 2023

$50m

SERIES C

Jul 2022

$30m

SERIES B

Total funding: $113.2m

Company benefits

Home office stipend: We invest in our people first and offer competitive salaries across all roles andEvery new team member gets U$1,000 to improve their home office. We want to make sure everyone has the setup to do their best work. locations
Equity: We want our team members to have skin in the game and an opportunity to see long-term returns on the hard work they are putting in. To that end, every team member receives a generous equity grant
Premium benefit offerings: We offer premium medical, dental, and vision plans and cover up to 100% of the premiums for both the employee and their family. We also offer opportunities to contribute to an HSA or FSA, short-term and long-term disability coverage, and life insurance
Unlimited PTO: We offer a flexible PTO policy to all team members and encourage everyone to take time off to recharge, do something fun, and spend time with loved ones
Team offsites: We have team and company-wide retreats in amazing locations to give everyone an opportunity to meet in person, collaborate, and build working relationships that make us a better team
Global remote team: We hire the best person for the job, regardless of location. Our remote-first culture allows team members to make an impact from anywhere in the world
Diversity and inclusion: We celebrate diversity and are committed to creating an inclusive environment for all employees. We are proud to be an equal-opportunity workplace
Flexible schedule: Strong teams are built on mutual trust and accountability. We believe in giving our team members ownership over their work and a flexible schedule to allow them to operate at their highest potential

Company HQ

Inner Sunset, San Francisco, CA

Leadership

Dylan Fox

(Founder & CEO)

Previously founded and acted as CEO for YouGive1. Experience as Senior Software Engineer for Cisco and as a Lead Research Assistant for NDP Consulting.

Share this job

View 2 more jobs at AssemblyAI