岗位职责
团队介绍:字节跳动基础设施部门,负责字节跳动的全栈算力基础设施,从芯片服务器产研、到超大规模数据中心,到传统云平台和如今的AI Native Cloud,全方位为抖音、豆包、今日头条、飞书、火山引擎等各类产品提供领先、稳定的百万量级大规模算力基础设施服务。我们的领域涵盖数据中心建设、内核操作系统开发、服务器集群管理、高速网络通信、EB级数据存储体系、搜索型数据库探索、基于LLM的AI基础设施的研发调度与治理等,致力于打造业界领先的、面向LLM的AI云原生基础设施架构与产品矩阵。
- 核心研发工作:
1)负责集合通信库(CCL)设计、开发与优化,支持多卡、多机场景下的高性能通信;
2)基于RDMA和scale up等互联技术,研发高速通信库及相关工具;
3)探索通信与计算的协同优化,设计通信算子与计算算子的融合方案;
- 性能优化与问题解决:
1)分析并优化AI集群中单机内、多机间的集合通信性能(如AllReduce、All2All等),突破软硬件性能瓶颈;
2)解决分布式训练、推理场景中的通信问题,适配复杂应用场景需求;
3)利用Triton等工具开发高性能通信计算融合算子,优化端到端通信延迟;
- 技术协作与生态建设:
1)与训练、推理、自研软硬件等团队协作,优化AI模型通信效率,支撑整体系统性能提升;
2)洞察集合通信技术趋势,推动自研通信库的架构演进与生态适配。
任职要求
- 2027届获得本科及以上学历,计算机、网络、软件工程等相关专业优先;
- 热爱技术,精通至少一门编程语言,包括但不仅限于:Java、C、C++、PHP、Python、Go;
- 有优秀的逻辑分析能力,能够对业务逻辑进行合理的抽象和拆分;
- 有强烈的求知欲,优秀的学习和沟通能力。