岗位职责
- 提升团队实验速度，加快实验启动、结果获取和迭代
- 维护并优化基于JAX的分布式训练框架，确保在TPU上的可扩展性和效率
- 分析模型数据质量，构建工具以发现并解决数据问题，转化为可衡量的准确率提升
- 优化生产环境中的语音语言模型推理，包括服务架构和量化、推测解码等高级技术
任职要求
- 精通JAX和TPU，熟悉Flax、Optax、XLA编译流程
- 具备测量纪律，能预先定义成功标准并对结果保持怀疑
- 有生产环境推理系统优化经验，最好涉及LLM或语音模型
- 深入理解大规模分布式训练、现代深度学习系统和ML基础设施最佳实践
- 熟悉现代推理优化技术：连续批处理、KV缓存管理、分片策略、量化
加分项
- 语音转文字领域知识：ASR架构、音频处理、流式推理
- C++或Rust经验用于内核级工作
其他说明
薪资范围$270,000-$310,000（美国地区），其他地区可能调整。面试过程中会使用AI工具。
JAX 远程职位概况
数字只统计当前仍在招聘的职位，与上方列表里已到期的条目不重复计入。
- 在招职位
- 1 个
- 累计收录
- 1 个
