The posting
团队介绍:AI芯片团队深耕人工智能芯片研发领域,产品覆盖云端与终端场景,致力于提供高效率、低成本的AI芯片及 SoC 解决方案,满足公司内外AI及相关业务的算力需求。团队自主设计了多款AI芯片,涵盖推理、训练场景,具有高性能、高精度、高灵活性、编译器友好、低成本等特点。核心AI加速核与AI软件栈均为自主研发,具备完整知识产权,能够支撑面向不同模型、不同部署场景的持续演进与优化。
1、负责自研硬件训练框架开发(Torch、Megatron、Dtensor等); 2、对接业务大模型例如豆包、Seedance等模型在自研硬件上训练任务支持; 3、开发DP、SP、TP、PP等分布式并行方式并且优化; 4、训练业务通信、计算、通算融合算子研发以及性能优化。
1、1-5年异构硬件、GPU相关领域训练开发经验,熟悉PyTorch、FSDP、Megatron-LM、VeRL分布式训练框架; 2、对数据并行、模型并行、分布式数据并行等常见训练模式有深入理解; 3、参与过硬件加速器训练相关研发工作,包含Torch Compiler、算子开发、编译调度优化等; 4、有主动学习、快速解决问题的能力和自我驱动力。
以下为加分项: 1、有大模型训练百、千卡以上集群训练落地经验; 2、有异构硬件、NPU等训练、推理开发经验,对硬件比较了解; 3、熟悉计算机体系结构和并行计算; 4、有GPU、FPGA或AI芯片相关的开发和评测经验。



