|
|
|
|
职位描述 用小程序查看更多 |
|
"职位描述: 我们正在寻找对人工智能系统底层优化充满热情的实习生加入团队!您将有机会参与到前沿AI加速技术的研究与开发中,在资深工程师的指导下,学习并实践如何为特定硬件平台(如GPU/AI加速卡)设计和优化高性能计算算子(Kernel),直接助力提升深度学习模型的运行效率。
实习职责: 参与算子库开发支持:在导师指导下,协助开发和测试用于深度学习模型的高性能计算算子(Kernel)。 辅助性能分析与优化:学习使用性能分析工具,协助定位模型在特定硬件(如GPU/NPU)上的计算瓶颈。 学习与探索优化技术:探索并尝试应用针对特定硬件架构(如CUDA核心、TensorCore、专用AI指令)的高性能编程和优化技术。 工具链与框架熟悉:熟悉并学习使用团队内部的算子开发工具链、测试框架以及相关的AI框架(如PyTorch,TensorFlow等)接口。 文档与测试:协助编写代码注释、开发文档,并参与算子功能的单元测试和性能测试。
任职要求: 扎实的编程基础:熟练掌握C或Python中的至少一种,具备良好的编程习惯和调试能力。 计算机基础:了解计算机体系结构、操作系统原理、数据结构和算法等基础 |
|
|
|
|
|
|