# Senior Research Engineer at Assembly AI

- Company: Assembly AI
- What the company does: Backed by Accel, Insight and Y Combinator.
- Company website: https://www.assemblyai.com/
- Type: Startups (AI role)
- Level: Senior
- Location: Remote
- Work setup: Remote
- Pay: $270K to $310K base salary per year (USD)
- Posted: 2026-09-01
- Apply by: 2026-10-16
- Apply: https://job-boards.greenhouse.io/assemblyai/jobs/4729671005
- Page: https://www.1752.vc/careers/jobs/assembly-ai-senior-research-engineer/

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

## What they're looking for

- Expert-level proficiency with JAX and TPUs , including the surrounding ecosystem (Flax, Optax, the XLA compilation pipeline)
- Measurement discipline. You define what success looks like before you start, you stay skeptical of your own results until they hold up, and you treat an unexplained improvement as a problem rather than a win
- Strong experience optimizing inference systems for production, ideally with LLMs or speech models
- Deep understanding of distributed training at scale, modern deep learning systems, and ML infrastructure best practices
- Familiarity with modern inference optimization techniques: continuous batching, KV-cache management, sharding strategies, quantization
- Enthusiasm for refactoring and improving existing systems — you thrive on making products and code faster and better

Tags: Research