高级研究工程师
Senior Research Engineer
#### **为什么选择AssemblyAI**
AssemblyAI打造了行业领先的语音AI模型，为下一代语音应用提供支持。我们的模型每月处理超过6亿次推理调用，每天处理超过100万小时的音频，并为20亿+终端用户提供体验。语音AI领域正处于转折点；我们正在寻找真正对加入小团队并帮助定义行业发展未来充满热情的人。
我们是地球上最高效利用资本的AI公司之一——员工人数不足100人，每位员工创造约50万美元的年度经常性收入，我们位列当今增长最快的AI公司中收入最密集的前五名团队。这不是偶然；这是有意保持精简、快速行动，并赋予每个团队成员巨大的所有权和影响力。拥有Granola、Fireflies、Figure AI和CallRail等数千家客户，公司已具备真正的规模——每天处理超过200万小时的音频，每天处理超过100万次API调用。这是一个罕见的成长阶段机会，业务已得到验证，发展轨迹陡峭，但团队仍足够小，你的影响会体现在每一件事上。
如果你曾因层层官僚主义而感到窒息，因缺乏真正的所有权而感到沮丧，或因自己的工作消失在缓慢运转的组织中而感到挫败，AssemblyAI的运作方式不同。公司作为一个真正的能者为王的环境运作，没有繁重的计划或审批流程，也没有对所需工具或信息的限制。对于任何真正关心语音AI的人，不是作为追逐趋势，而是作为构建技术，这里是解决最有趣问题、在最有趣的规模下工作的场所，团队小到你真的会认识每个人的姓名。
我们致力于创造一个空间，让员工能够带真实的自我来工作，并享有平等的成功机会。无论你的种族、性别认同或表达、性取向、宗教、出身、能力、年龄、退伍军人身份如何，如果加入这个使命触动了你，我们鼓励你申请！
### **职位描述**
我们正在寻找一名**高级研究工程师**加入我们的研究团队，负责开发和改进大规模分布式训练、数据处理和推理系统。我们公司的目标是通过模型开发和测量快速解决客户问题并提升产品性能——我们前进的速度取决于这里任何人运行实验的速度。
查看英文原文
#### **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 $500K 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 our models learn from: investigating quality issues, building the tooling to surface them, and turning what you find into measurable accuracy gains.
- Analyze the accuracy of production models, build evaluation harnesses, and work out which improvements will matter most to customers.
- Translate research prototypes into production-ready systems, refactoring and modernizing model architectures and infrastructure along the way.
- Optimize production inference for speech language models, both from a serving architecture perspective and through advanced techniques such as quantization and speculative decoding.
- Investigate and resolve performance bottlenecks across the stack, from low-level kernels (XLA, Pallas) to high-level system design.
- Partner with researchers, infrastructure, and production engineering to trace problems to their real source and ship fixes that hold.
### **What You’ll Need**
- **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.
- **Appetite for the whole pipeline.** Your core strength might be JAX and TPU performance, but when a customer issue traces back to a data problem or an evaluation blind spot, you want to go find it yourself. The people who do well here went deep in one area first, then kept expanding outward.
- 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.
- Strong Python skills; C++ or Rust experience for kernel-level work is a plus.
- Excellent communication and a collaborative mindset — you can clearly explain complex tradeoffs and prioritize high-impact work.
### **Bonus**
- **Domain knowledge in Speech-to-Text:** ASR architectures, audio processing, streaming 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.
Salary range: $270,000 - $310,000
#### **AI to Interview:**
If you’re selected for an interview, please review this [resource](https://www.assemblyai.com/candidate-ai-guidance) to better understand how AssemblyAI approaches the use of AI in our interview process.
#### **GDPR privacy notice:**
Candidates from the EU should review [this job applicant privacy notice](https://www.assemblyai.com/legal/gdpr-job-applicant-and-worker-privacy-notice) before applying.
#### **Keep Exploring AssemblyAI:**
[Speech-to-text](https://www.assemblyai.com/products/speech-to-text) | [Streaming speech-to-text](https://www.assemblyai.com/products/streaming-speech-to-text) | [Speech Understanding](https://www.assemblyai.com/products/speech-understanding) | [LLM Gateway](https://www.assemblyai.com/products/llm-gateway) [Try the Playground](https://www.assemblyai.com/playground/source)
[Our $50M Series C fundraise](https://www.assemblyai.com/blog/announcing-our-50m-series-c-to-build-superhuman-speech-ai-models/)
[Check us out on YouTube!](https://www.youtube.com/@AssemblyAI)