Senior Research Engineer

Headquarters: Remote Why AssemblyAI AssemblyAI builds the best-in-class Voice AI models powering the next generation of voice applications. Our models serve 600M+ inference calls monthly, process 1M+ hours of audio daily, and power 2 billion+ end-user experiences. The Voice AI space is at an inflection point; we’re looking for folks truly excited to join a small team and help define the future of the industry. We are one of the most capital-efficient AI companies on the planet - with under 100 people generating roughly $600K ARR per employee, we sit among the top 5 most revenue-dense teams within the fastest-growing AI companies today. That's not an accident; it's a deliberate choice to stay lean, move fast, and give every person on the team outsized ownership and impact. With thousands of customers including Granola, Fireflies, Figure AI, and CallRail, the company has real scale - processing over 2 million hours of audio daily and handling more than 1 million API calls every day. This is a rare growth-stage opportunity where the business is proven and the trajectory is steep, but the team is still small enough that your fingerprints are on everything. If you've ever felt buried under layers of bureaucracy, starved of real ownership, or frustrated watching your work disappear into a slow-moving org, AssemblyAI is built differently. The company operates as a true meritocracy, with no heavy planning or approval processes and no gatekeeping on the tools or information you need. For anyone who genuinely cares about voice AI, not as a trend to chase, but as a technology to build, this is the place where the most interesting problems at the most interesting scale are being solved by a team small enough that you'll actually know everyone's name. We’re committed to creating a space where our employees can bring their full selves to work and have equal opportunity to succeed. No matter your race, gender identity or expression, sexual orientation, religion, origin, ability, age, veteran status, if joining this mission speaks to you, we encourage you to apply! About the Role We're looking for a Senior Research Engineer to join our Research team, developing and improving the systems behind large-scale distributed training, data processing, and inference. Our goal as an organization is to solve customer problems and improve our products quickly through model development and measurement — and how fast we move depends on how quickly anyone here can run an experiment, measure it, and find out what's wrong. Raising that ceiling is the heart of this role. You'll be working inside the pipeline you're improving, not alongside it. The ideal candidate has a deep understanding of modern deep learning systems, combined with strong engineering expertise across JAX and TPUs, layer-level optimization, large-scale distributed training, streaming, low-latency and asynchronous inference, inference compilers, and advanced parallelization techniques. This is a cross-functional role. You'll work closely with our researchers, our infrastructure team, and production engineering — not as a handoff point, but as the person who learns enough of each domain to follow problems through to resolution. At times you'll train models, run evaluations, and analyze data yourself, both to deliver impact directly and to learn what's worth building to multiply the team's work. The bar is someone who understands the end-to-end impact they intend to make, measures it from the outset, and would rather find out they were wrong in a week than in a quarter. That discipline is what turns cross-functional ownership into an advantage. The role is embedded within the Research team. What You’ll Do Raise the team's experimental velocity — make it faster to launch an experiment job, get a number back you can trust, and know what to try next. Maintain and evolve our JAX training framework, keeping it scalable and efficient for large-scale distributed training runs on TPU. Improve the data o