岗位职责
- 负责电商LLM/VLM/AIGC训练优化算法研究（和算法团队合作）和落地；
- 通过数据并行、模型并行、Pipeline并行、通信优化等基于CUDA的分布式训练框架大幅提升模型的训练速度与效率；
- 攻克电商LLM/VLM/AIGC训练全流程中工程技术问题，包括CT、SFT、RL等多个阶段，覆盖性能、资源、调度、平台化等能力；
- 与公司各算法部门深度合作，分析业务性能瓶颈，通过软硬结合的方式，高效部署与优化业务模型，落地到电商各产品线，投入AI工具链开发及技术生态的建设，支撑电商AI方向重要业务的发展。
职位要求
- 本科及以上学历，计算机、软件工程、人工智能等相关专业优先，熟练掌握Linux环境下的C/C++或Python语言；
- 了解深度学习算法基本原理，熟悉神经网络基本架构和各算子计算方式，1年以上AutoML、模型推理/训练优化或AI相关工作经验；
- 能够熟练使用至少一种主流的机器学习框架（TensorFlow/PyTorch/MxNet或其他自研框架），熟悉各种模型/数据并行训练框架优先；
- 了解主流LLM/VLM/AIGC算法模型，有LLM/VLM/AIGC模型训练加速落地经验者优先；
- 有大模型训练优化经验，熟悉FSDP/DeepSpeed/Megatron等分布式训练框架经验者优先；
- 具有独立解决问题的能力，良好的团队合作精神，有强烈的工作责任心，较好的学习能力、沟通能力和自驱力。
