电商场景LLM/VLM/AIGC训练工程师-TikTok Shop

职位描述

1、负责电商LLM/VLM/AIGC训练优化算法研究（和算法团队合作）和落地；

2、通过数据并行、模型并行、Pipeline并行、通信优化等基于CUDA的分布式训练框架大幅提升模型的训练速度与效率；

3、攻克电商LLM/VLM/AIGC训练全流程中工程技术问题，包括CT、SFT、RL等多个阶段，覆盖性能、资源、调度、平台化等能力；

4、与公司各算法部门深度合作，分析业务性能瓶颈，通过软硬结合的方式，高效部署与优化业务模型，落地到电商各产品线，投入AI工具链开发及技术生态的建设，支撑电商AI方向重要业务的发展。

任职要求

1、本科及以上学历，计算机、软件工程、人工智能等相关专业优先，熟练掌握Linux环境下的C/C++或Python语言；

2、了解深度学习算法基本原理，熟悉神经网络基本架构和各算子计算方式，1年以上AutoML、模型推理/训练优化或AI相关工作经验；

3、能够熟练使用至少一种主流的机器学习框架（TensorFlow/PyTorch/MxNet或其他自研框架），熟悉各种模型/数据并行训练框架优先；

4、了解主流LLM/VLM/AIGC算法模型，有LLM/VLM/AIGC模型训练加速落地经验者优先；

5、有大模型训练优化经验，熟悉FSDP/DeepSpeed/Megatron等分布式训练框架经验者优先；

6、具有独立解决问题的能力，良好的团队合作精神，有强烈的工作责任心，较好的学习能力、沟通能力和自驱力。