6 дней назад
Senior Research Engineer (AI)
270 000 - 310 000$
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Senior Research Engineer (JAX/TPU): Developing and improving distributed training, data processing, and low-latency inference systems for large-scale Voice AI models with an accent on JAX and TPU performance, evaluation, and production inference optimization. Focus on building scalable training frameworks, optimizing speech language model serving with quantization and speculative decoding, and resolving bottlenecks from low-level kernels to high-level system design.
Location: Remote - New York; salary range applies to candidates in the U.S.
Salary: $270,000–$310,000 per year
Company
develops Voice AI models for speech applications, processing large-scale audio and inference workloads for thousands of customers.
What you will do
- Increase experimental velocity by improving experiment execution, measurement, and debugging workflows.
- Maintain and evolve the JAX training framework for scalable distributed training on TPUs.
- Investigate training-data quality issues, build tooling to surface them, and convert findings into measurable accuracy improvements.
- Analyze production-model accuracy, develop evaluation harnesses, and identify improvements that matter to customers.
- Translate research prototypes into production systems while refactoring model architectures and infrastructure.
- Optimize production inference for speech language models and resolve performance bottlenecks across kernels, compilers, and system design.
Requirements
- Expert-level proficiency with JAX, TPUs, Flax, Optax, and the XLA compilation pipeline.
- Strong experience with production inference optimization, preferably for LLMs or speech models.
- Deep understanding of distributed training at scale, modern deep learning systems, and ML infrastructure.
- Familiarity with continuous batching, KV-cache management, sharding strategies, and quantization.
- Strong Python skills and the ability to communicate complex technical trade-offs across research, infrastructure, and production engineering.
- Experience with C++ or Rust for kernel-level work is a plus.
Nice to have
- Speech-to-Text domain knowledge, including ASR architectures, audio processing, and streaming inference.
Culture & Benefits
- Small, lean team with substantial individual ownership and impact.
- Meritocratic environment with limited bureaucracy and direct access to tools and information.
- Equal-opportunity workplace focused on inclusion and diverse backgrounds.
- Competitive compensation and pay-equity practices.
Hiring process
- Selected candidates should review 's guidance on the use of AI in interviews.
- Salary may be adjusted for candidates outside the U.S. and based on experience, qualifications, and interview assessment.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
4 дня назад
Staff Machine Learning Engineer (AI/LLM)
195 000 - 233 800$
3 дня назад
Senior AI Engineer (Speech)
224 500 - 256 000$
Swayable
4 дня назад
Senior Engineer (AI)
175 000 - 250 000$
7 дней назад
Senior Software Engineer (Autonomous Driving)
245 000 - 285 000$
9 часов назад
Member of Technical Staff, Machine Learning (AI)
6 дней назад
Senior AI/ML Engineer (Aerospace)
143 487 - 197 295$