Senior Software Engineer, Machine Learning

permanent
Fully Remote

Only accepting applications from: United States

  • 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 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.

Experience

  • 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
  • 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

Salary and Perks

Pay range: $195K - $225K

About AssemblyAI

Industry-leading Speech AI models to automatically recognize and understand speech.

Industry-leading Speech AI models to automatically recognize and understand speech.

View all developer jobs

Workster

Remote Jobs for US Residents

We've built a new platform specifically for US residents to find remote work.

Discover Workster

Power Search

Find the jobs that don't get advertised

We've built a tool to help you discover all of the remote jobs that never get advertised.

Discover Power Search