Senior Software Engineer, AI Data

Join to apply for the Senior Software Engineer, AI Data role at AssemblyAI

At Assembly

AI, 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, Assembly

AI 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 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 help improve childhood literacy via ed‑tech tools.

We’ve raised funding from 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!

We’re seeking an exceptional Senior Software Engineer to join our AI Data team. This role is focused on building robust, scalable systems that power our AI data platform. You’ll work on high‑impact projects that directly influence our ability to train, evaluate, and deploy models at scale, with a strong emphasis on software engineering excellence, system reliability, and code quality.

As a Senior Engineer, you’ll drive technical execution within your team, taking ownership of significant features and components. You should be passionate about writing clean, maintainable code, implementing comprehensive testing strategies, and continuously improving engineering practices. This role requires close collaboration with researchers, platform engineers, and other stakeholders. You’ll need to balance technical excellence with pragmatic delivery in a fast‑paced startup environment.

• Design scalable, future‑proof data platforms optimized for AI research workloads
• Build efficient self‑serve data processing pipelines leveraging GCP’s advanced services
• Implement cost‑effective storage and monitoring solutions for ML at scale
• Create flexible training resource management with intelligent queuing
• Optimize resource allocation for maximum training efficiency
• Participate in on‑call rotation to ensure system reliability

• Lead adoption of cutting‑edge ML tools and frameworks, continuously evaluating and integrating best‑in‑class solutions
• Streamline existing workflows while introducing new tooling that further reduces complexity
• Enhance our tooling and documentation to accelerate team velocity and maintain our competitive edge
• Implement guardrails for cost, quality, and performance
• Identify and eliminate technical bottlenecks in the data processing and training pipelines

• 5+ years of professional software engineering experience
• Strong proficiency in Python and SQL with demonstrated ability to write production‑quality code
• Solid understanding of software engineering fundamentals:
  • Data structures and algorithms
  • System design and architectural patterns
  • Testing strategies (unit, integration, end‑to‑end)
  • Code review practices and technical collaboration
• Experience with:
  • RESTful APIs and distributed systems concepts
  • Containerization (Docker) and basic cloud infrastructure
• Track record of delivering high‑quality software in a team environment
• Ability to thrive in a startup environment with changing priorities and rapid iteration

• Experience with GCP services (Big Query, GCS, Cloud Run, GKE)
• Familiarity with distributed processing frameworks (Apache Beam, PySpark)
• Experience with…