技术团队成员，智能体环境
岗位摘要
设计并编写高性能、可扩展的模型训练软件；开发新工具以支持并加速研究与 LLM 训练；与其他工程团队（基础设施、效率、服务）及科学团队（智能体、多模态、多语言等）协调合作；在集群和数据基础设施上研究、实现并实验新想法
岗位职责
- 设计并编写高性能、可扩展的模型训练软件
- 开发新工具以支持并加速研究与 LLM 训练
- 与其他工程团队（基础设施、效率、服务）及科学团队（智能体、多模态、多语言等）协调合作
- 在集群和数据基础设施上研究、实现并实验新想法
- 与科学家、工程师及各团队保持密切协作
任职要求
- 重视测试驱动开发、整洁代码，并致力于在各层面减少技术债务
- 精通 Python 及相关机器学习框架，如 JAX、PyTorch 和/或 XLA/MLIR
- 具有使用和调试大规模分布式训练策略（内存/速度性能分析）的经验
- 具有分布式训练基础设施（Kubernetes）及相关框架（Ray）的经验
- 具备机器学习和 LLM 学术研究经验
福利待遇
- 每周提供 75 美元/75 英镑或等值当地货币的午餐补贴
- 全面的健康与牙科福利，包括专门的心理健康预算
- 提供 RRSP 匹配、401K、养老金计划
- 100% 育儿假