蚂蚁集团招聘异构计算研发工程师，工作地点北京/杭州

蚂蚁集团社会招聘异构计算研发工程师，岗位设在北京和杭州，要求计算机相关专业、3 年以上相关工作经验。职责涵盖 GPU 等 AI 芯片的软硬件协同设计、大模型训推引擎优化，以及与算法团队配合的系统级优化落地。任职要求包括掌握 Megatron-LM、DeepSpeed、PyTorch FSDP 等框架及 CUDA 编程模型，有千卡级集群训练、MoE 预训练经验者优先。