小米训练平台性能优化工程师实习生
实习兼职地点：北京状态：招聘♡ 收藏
工作描述
任职要求 1.精通深度学习训练加速技术，熟悉分布式训练框架设计 2.掌握CUDA编程及GPU性能分析工具（nsys/torch profiler） 3.熟练使用PyTorch框架，具备训练流程优化实战经验 4.熟悉常见模型压缩技术（量化/剪枝/蒸馏）及落地应用 5.具备大规模集群资源调度系统开发经验者优先 【技术加分项】 1.有mmcv//deepspeed/me…
登录查看完整工作描述
微信扫码，1秒登录
包括英文材料
深度学习+
https://d2l.ai/
Interactive deep learning book with code, math, and discussions.
CUDA+
https://developer.nvidia.com/blog/even-easier-introduction-cuda/
This post is a super simple introduction to CUDA, the popular parallel computing platform and programming model from NVIDIA.
https://www.youtube.com/watch?v=86FAWCzIe_4
Lean how to program with Nvidia CUDA and leverage GPUs for high-performance computing and deep learning.
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.
DeepSpeed+
[英文] Getting Started
https://www.deepspeed.ai/getting-started/
DeepSpeed model training is accomplished using the DeepSpeed engine.
https://www.youtube.com/watch?v=pDGI668pNg0
Megatron+
https://www.youtube.com/watch?v=hc0u4avAkuM
还有更多 •••
相关职位
实习
1、计算机相关专业本科及以上学历在读，具备扎实的计算机理论基础； 2、熟悉Python/C++/Golang等其中一种编程语言，有熟练的Shell脚本编写技能； 3、有强烈的上进心，自我驱动，学习适应
更新于 2025-10-14武汉
社招2年以上程序&技术类
1.本科及以上学历，两年及以上经验，精通 Python，能读写 Go 或 C++ 之一 下面两项至少一项有实战，另一项能看懂； 2.读过并改过 verl、OpenRLHF、NeMo-RL 之一的源码，
上海
社招2年以上程序&技术类
1.本科及以上学历，两年及以上经验，精通 Go； 2.独立设计并上线过至少一个生产级 Operator（kubebuilder 或 controller-runtime），讲得清 reconcile
上海
实习软件序列
基本要求 - 计算机科学、人工智能或相关方向 在读博士 - 扎实的编程基础，熟练掌握 Python 或 C++ - 熟悉至少一种深度学习框架，例如 PyTorch - 熟悉 Linux开发环境，具备良
更新于 2026-03-16北京