字节跳动 社会招聘
电商场景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、具有独立解决问题的能力,良好的团队合作精神,有强烈的工作责任心,较好的学习能力、沟通能力和自驱力。
官网发布:2025-03-25 · 最后确认在招:2026-09-03 19:35:21 · 来源平台:feishu