
商汤|大装置-AI训练/推理框架研发实习生
岗位职责
实习表现优异,有实习转正机会!岗位可根据兴趣选择侧重以下方向之一:
方向一:训练框架开发与优化
1. 参与深度学习框架的代码开发与优化,支持大模型训练与微调,提升系统性能与稳定性。
2. 协助优化框架核心功能与计算能力,参与系统架构与模型设计的改进工作。进行性能优化,包括计算优化、通信优化及调用流程优化等。
3. 参与大模型训练相关技术方案的讨论、文档撰写与架构设计支持。
4. 开展国内外相关技术与产品调研,协助探索创新方向。
5. 参与公司内的商业化交付和政府项目交付,主要内容是训练软件与芯片通用训练支持。
方向二:推理框架开发与优化
1. 参与大模型推理框架的开发与优化,提升推理效率与系统稳定性。
2. 协助扩展框架功能与计算能力,参与系统架构与模型设计的优化工作。进行性能优化,包括计算优化、通信优化及调用流程优化等。
3. 参与大模型推理相关技术方案的沟通、规划与文档支持。
4. 开展相关技术调研,协助跟踪业界进展并探索创新路径。
5. 参与公司内的推理API服务部署服务优化。
任职要求
1. 本科及以上学历优先,计算机专业基础知识扎实:含数据结构,操作系统,组成原理等。
2. 熟悉C/C++编程,有python使用经验优先。有代码调试、debug能力,有良好的代码风格和编程习惯。
3. 对于主流Deep Learning框架PyTorch、TensorFlow等有了解,熟悉其内部架构或实现机制者优先。
4. 有较强的团队精神和沟通交流能力。
5. 每周至少到岗4天,能够实习6个月优先。
加分项:
1. 参与过开源项目优先。
2. 有CUDA编程经验优先。
商汤招聘实习生:大装置-AI训练/推理框架研发实习生岗位来自商汤招聘官网,具体内容以商汤招聘官网为准。
