字节跳动
社招 · 正式电商场景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、具有独立解决问题的能力,良好的团队合作精神,有强烈的工作责任心,较好的学习能力、沟通能力和自驱力。
内推申请说明
本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道,不等于已在公司官网投递;薪资、岗位状态和实际招聘流程以官方信息为准。