小米训练平台性能优化工程师实习生
实习兼职地点：北京状态：招聘
工作描述
任职要求 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+
https://www.youtube.com/watch?v=pDGI668pNg0
Megatron+
https://www.youtube.com/watch?v=hc0u4avAkuM
还有更多 •••
相关职位
实习软件序列
基本要求 - 计算机科学、人工智能或相关方向 在读博士 - 扎实的编程基础，熟练掌握 Python 或 C++ - 熟悉至少一种深度学习框架，例如 PyTorch - 熟悉 Linux开发环境，具备良
更新于 2026-03-16北京
校招软件序列
基本要求 - 计算机科学、人工智能或相关方向 在读博士 - 扎实的编程基础，熟练掌握 Python 或 C++ - 熟悉至少一种深度学习框架，例如 PyTorch - 熟悉 Linux开发环境，具备良
更新于 2026-03-16北京
校招
岗位简介 加入我们，你将参与国内头部自动驾驶公司万卡级GPU训练集群的构建与优化，直接支撑端到端自动驾驶大模型的训练迭代。在这里，你会深入PyTorch编译器、CUDA Kernel、NCCL通信库的
更新于 2026-07-07深圳|广州|北京
社招2年以上软件序列
1. 本科及以上学历，计算机、软件工程、人工智能、通信、自动化等相关专业优先。 2. 具备 2年以上技术支持、项目交付、测试工程师、运维工程师或AI平台相关工作经验，有企业客户交付经验优先。 3. 熟
更新于 2026-07-06北京