Senior Research Engineer
Core
Develop and improve systems for large-scale distributed training, data processing, and inference for Voice AI models.
Role type
Senior Research Engineer (ML Infrastructure)
Builds
Scalable JAX training frameworks, production inference systems, and evaluation harnesses for speech models.
Domain
AI/ML Infrastructure, Voice AI, Distributed Systems
Deliverable
production ML models | infrastructure
Required skills
JAX, TPUs, distributed training, deep learning systems, inference optimization, Python, system design
Preferred skills
C++ or Rust, speech-to-text (ASR), audio processing, streaming inference, quantization, speculative decoding
Technologies
JAX, Flax, Optax, XLA, Pallas, TPU
Responsibilities
Maintain and evolve JAX training framework for large-scale distributed runs; improve data quality and build tooling to surface issues; analyze production model accuracy and build evaluation harnesses; translate research prototypes into production-ready systems; optimize production inference via quantization and speculative decoding; investigate and resolve performance bottlenecks across the stack.
Seniority
Senior, hands-on IC
