腾讯混元大模型训练框架研发工程师-(北京/深圳)
社招全职5年以上公共技术地点:北京状态:招聘♡ 收藏
工作描述
任职要求
1.熟练使用PyTorch框架,可对DDP训练的代码进行性能分析和优化;
2.熟练使用主流大模型训练框架DeepSpeed、Megatron,掌握3D并行、ZeRO机制、Flash-Attn等的原理、使用场景、优劣势以及可优化方向;
3.有ViT、S…登录查看完整工作描述
微信扫码,1秒登录
包括英文材料
PyTorch+
https://datawhalechina.github.io/thorough-pytorch/
PyTorch是利用深度学习进行数据科学研究的重要工具,在灵活性、可读性和性能上都具备相当的优势,近年来已成为学术界实现深度学习算法最常用的框架。
https://www.youtube.com/watch?v=V_xro1bcAuA
Learn PyTorch for deep learning in this comprehensive course for beginners. PyTorch is a machine learning framework written in Python.
还有更多 •••
相关职位
社招3年以上AI技术
1.熟悉常用的大模型训练推理框架,如PyTorch/Megatron/VLLM/Verl,有相关开发经验; 2.熟悉Docker、Kubernetes等容器和容器编排技术; 3.熟悉编译器相关技术
更新于 2026-04-03深圳
社招3年以上混元-模型算法技
1.计算机科学/软件工程/人工智能等相关专业硕士及以上学历; 2.深入理解 Transformer 架构及大语言模型训练原理,在 LLM Alignment、RLHF、Reward Modeling
更新于 2026-06-30深圳