工作职责 • 研究业界最新的分布式通信方案和并行计算方案 • 设计/开发高性能、低延时、易用、可靠的分布式通信和并行计算组件 • 分布式通信和并行计算组件性能调优 • 集群环境下大模型推理和训练的系统调优 • 参与下一代人工智能芯片的互联架构设计

职位基本要求 • 计算机，电子工程，数学等相关专业硕士及以上学历，特别优秀者可以放宽到本科学历。 • 积极主动，有责任心；学习能力强，对问题有钻研精神；沟通顺畅，团队合作意识好 • 有扎实的计算机基础和丰富的软件工程经验，了解AI相关基础知识 • 熟悉Linux下的C/C++软件开发，熟悉cuda编程 职位要求加分项 • 有集群环境下大模型开发、调优、部署相关经验者 • 有大模型框架和工具（如megatron、deepspeed、fairscale等）经验，熟悉各类并行切分算法 • 有并行计算和通信组件（如OpenMPI、NCCL、NVSHMEM等）相关经验者 • 擅长GPU、虚拟化、存储、编译器、HPC等其中一个或多个领域 • 擅长linux驱动和内核开发 • 熟练使用性能分析工具（如perf、vtune、gperftools等），擅长性能调优