Software Engineer, Machine Learning

1 Month ago • All levels • Research Development • $157,500 PA - $175,000 PA

Job Summary

Job Description

AssemblyAI is building at the forefront of Speech AI, creating powerful models for speech-to-text and speech understanding. They are seeking a Machine Learning Engineer to accelerate their AI research-to-production pipeline. This role involves building infrastructure for rapid model deployment, safe testing, and efficient, scalable production inference systems. The ideal candidate will have a strong backend engineering background in distributed systems and containerization, with a deep interest in optimizing the path from research innovation to production value. This is a cross-functional role requiring close collaboration with research and engineering teams. The company is a remote team funded by leading investors and is focused on becoming a great AI company.
Must have:
  • Strong backend engineering experience with Python
  • Experience building/operating distributed, containerized apps on AWS
  • Proficiency in observability solutions
  • Ability to design resilient, scalable architectures
Good to have:
  • MLOps experience with PyTorch and Kubernetes
  • Startup environment experience
  • Experience with globally distributed teams
  • Comfort across the ML lifecycle
  • Experience in audio-related domains
  • Experience with other cloud providers
  • Familiarity with Ray.io, Bazel, monorepos
  • Experience with alternative ML inference frameworks
  • Experience optimizing for low-latency inference

Job Details

About AssemblyAI

At AssemblyAI, we’re building at the forefront of Speech AI, creating powerful models for speech-to-text and speech understanding available through a straightforward API. With more than 200,000 developers building on our API and over 5,000 paying customers, AssemblyAI is helping unlock and support the next generation of powerful, meaningful products built with AI. 

Progress in AI is moving at an unprecedented pace– and our team is made up of experts in AI research that are focused on making sure that our customers are able to stay on the cutting edge, with production-ready AI models that are constantly updating and improving as our team continues to improve accuracy, latency, and what’s possible with Speech AI. Our models consistently rank highest in industry benchmarks for accuracy, outperforming models from Google and Amazon, and up to 30% fewer hallucinations than OpenAI’s Whisper. Our models power more than 2 billion end-user experiences each day, helping companies better understand customer feedback, run more productive meetings with automated meeting notes, and helping improve childhood literacy via ed tech tools. 

We’ve raised funding by leading investors including Accel, Insight Partners, Y Combinator’s AI Fund, Patrick and John Collision, Nat Friedman, and Daniel Gross. We’re a remote team looking to build one of the next great AI companies, and are looking for driven, talented people to help us get there!

About the role:

We're looking for a Machine Learning Engineer to accelerate our AI research-to-production pipeline. This person will build infrastructure enabling our research team to rapidly deploy and safely test new models while maintaining efficient, scalable production inference systems. This person should have a strong backend engineering background in distributed systems and containerization, and be deeply interested in optimizing the path from research innovation to production value. This is a cross-functional role that requires close collaboration with both research teams developing models and engineering teams supporting the broader platform.

What You’ll Do:

  • Design and implement tooling that enables researchers to quickly deploy and evaluate new models in production 
  • Build and maintain high-performance, cost-efficient inference pipelines in production
  • Optimize infrastructure for both 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 complex production issues across distributed systems
  • Continuously improve our MLOps practices to reduce friction between research and production

What You’ll Need:

  • Strong backend engineering experience with Python
  • Experience building and operating distributed, containerized applications, preferably on AWS 
  • Proficiency implementing observability solutions (monitoring, logging, alerting) for production systems
  • Ability to design and implement resilient, scalable architectures

An ideal candidate should also have some of the following:

  • MLOps experience, including familiarity with PyTorch and Kubernetes
  • Experience working in startup environments demonstrating ownership, decisiveness, and rapid iteration
  • 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 Ray.io, Bazel, and monorepos
  • Experience with alternative ML inference frameworks beyond PyTorch
  • Experience optimizing for low-latency, real-time inference

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, benefit, and other reward opportunities we provide.

There are many factors that go into salary determinations, including relevant experience, skill level, 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 or less experienced than outlined in the job description. In this case, we will communicate any updates in the expected salary range.

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.

Similar Jobs

Zuora - Product Manager - Enterprise Solutions Integrations

Zuora

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
cirrus logic - Staff Embedded Software Engineer

cirrus logic

Taipei City, Taiwan (Hybrid)
2 Months ago
Qualcomm - WLAn Phy RTL Design Engineer - Staff

Qualcomm

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Zscaler - Product Support Engineer III - DLP

Zscaler

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
Qualcomm - DDR/ PCIE High Speed Interface IP Physical Design Engineer (Sr. Lead/Staff)

Qualcomm

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Ion - Research Editor

Ion

Mumbai, Maharashtra, India (On-Site)
4 Months ago
Ramboll3 - Lead Machine Learning Engineer

Ramboll3

Gurugram, Haryana, India (Hybrid)
2 Months ago
Ansys - R&D Engineer II

Ansys

Hsinchu County, Taiwan (On-Site)
1 Month ago
Inkittt - Senior Machine Learning Engineer, Recommendations

Inkittt

San Francisco, California, United States (Hybrid)
6 Months ago
Autodesk - Machine Learning Engineer

Autodesk

Montreal, Quebec, Canada (Hybrid)
1 Year ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Morning Star - Senior Data Research Analyst

Morning Star

Mumbai, Maharashtra, India (Hybrid)
2 Months ago
Ubisoft - Community Developer

Ubisoft

Paris, Île-de-France, France (Hybrid)
2 Months ago
PwC - Java Software Engineer

PwC

Bucharest, Bucharest, Romania (On-Site)
2 Months ago
Capgemini - Data Engineer

Capgemini

Kolkata, West Bengal, India (On-Site)
2 Months ago
Pingle studios - Art Director

Pingle studios

(Remote)
9 Months ago
London stock Exchange - INDEX POLICY SENIOR MANAGER, EQUITY, APAC

London stock Exchange

Hong Kong (On-Site)
2 Months ago
Scopely - Senior Product Manager, Performance - Monopoly GO!

Scopely

United States (Remote)
4 Months ago
SparkCognition - Software Engineer (Scala_Backend)

SparkCognition

Bengaluru, Karnataka, India (On-Site)
10 Months ago
Tencent - Business Development Manager

Tencent

Palo Alto, California, United States (On-Site)
2 Months ago
Demandbase - Senior Product Designer

Demandbase

San Francisco, California, United States (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

Jobs in United States

Nintendo - Sr BI Data Analytics Engineer

Nintendo

Redmond, Washington, United States (On-Site)
3 Months ago
Ettain Group - Senior QA Tester

Ettain Group

Westlake, Texas, United States (On-Site)
10 Years ago
Clearwater Analytics - Associate General Counsel

Clearwater Analytics

Boise, Idaho, United States (On-Site)
1 Year ago
Nordson Corporation - Supervisor, Production

Nordson Corporation

St. Petersburg, Florida, United States (On-Site)
2 Months ago
Blinkhealth - Patient Care Coordinator

Blinkhealth

Chesterfield, Missouri, United States (On-Site)
2 Months ago
Addepar - IT Support Specialist

Addepar

New York, United States (On-Site)
3 Weeks ago
Next Level Business Services - Teradata Developer

Next Level Business Services

St. Louis, Missouri, United States (On-Site)
9 Months ago
Instawork - Product Operations Manager

Instawork

Chicago, Illinois, United States (On-Site)
1 Month ago
Epic Games - Senior C++ Engineer

Epic Games

United States (On-Site)
3 Months ago
Toast - Bilingual Spanish Customer Growth District Manager

Toast

Miami, Florida, United States (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Hedra - Applied Research Scientist

Hedra

New York, New York, United States (On-Site)
4 Months ago
Apple - Principal Research Scientist

Apple

Cupertino, California, United States (On-Site)
2 Months ago
IMC - Machine Learning Engineer

IMC

Chicago, Illinois, United States (On-Site)
3 Months ago
Qualcomm - Machine Learning for Video Compression - Principal Scientist

Qualcomm

San Diego, California, United States (On-Site)
2 Months ago
Riot Games - Senior User Researcher

Riot Games

Dublin, County Dublin, Ireland (On-Site)
8 Months ago
bytedance - Machine Learning Engineer - MLDev

bytedance

San Jose, California, United States (On-Site)
4 Months ago
Nordson Corporation - Work student AI Software Engineer

Nordson Corporation

Eckental, Bavaria, Germany (On-Site)
1 Month ago
Apple - Machine Learning Engineer, GenAI & LLM

Apple

Sunnyvale, California, United States (On-Site)
2 Months ago
bytedance - Research Scientist Intern (Doubao (Seed) - Music Foundation Model) - 2024 Summer (PhD)

bytedance

San Jose, California, United States (On-Site)
9 Months ago
Adobe - Machine Learning Engineer 5 - AI Foundation - Image domain

Adobe

Bengaluru, Karnataka, India (On-Site)
1 Year ago

Get notifed when new similar jobs are uploaded