机器学习平台开发实习生

• 与算法团队紧密合作，深度参与大模型的分布式训练与推理加速优化工作，设计并实现高效的并行训练与推理架构； • 平台工具链建设与完善：参与公司自研 AI 训练和推理框架的设计与研发； • 前沿技术调研与探索：聚焦 性能优化、AI 推理、智能体（Agent）、RAG 等前沿技术方向，推动技术在平台产品中的落地与创新，持续提升平台竞争力。

• 本科及以上学历，熟悉常用深度学习算法和模型； • 熟练掌握 Python 和 C++，熟悉 FDSP、DeepSeed、Megatron 等分布式训练框架，熟悉至少一种深度学习框架 (如 Pytorch)； • 对复杂系统编程有扎实的理解和认识，如多线程、内存、网络、IO、编译等。能准确定位平台、系统、程序的性能问题，做出针对性改进； • 熟悉GPU/AI DSA 体系结构和原理，有 NVIDIA GPU 或国产 AI 芯片上的开发优化经验； • 熟悉 LLM 推理框架，有 transformer 推理加速方面经验。