职位&公司对比

职位详情

• 北京
• 1-3年
• 本科

• Python
• CUDA
• c++

工作职责：BOSS直聘 1.基于 pytorch 优化模型多机多卡训练效率； 2.在不同平台上部署深度学习模型，包括 CUDA，海思，amba，TI 等，和深度学习算法工程师配合，实现深度学习算法的软硬件联合优化以及深度学习相关工具链的开发； 3.组织编写相关文档，跨部门协作提供技术支持； 任职资格： 1.熟练掌握 C++、python 中的一种，另一种也需要了解，两种都kanzhun精通的优先； 2.熟悉常见深度学习模型，算子，了解 ONNX； 3.熟悉深度学习模型分布式训练或者不同平台的模型量化部署流程和细节，有分来自BOSS直聘布式训练优化经验或模型量化性能调优经验的优先； 4.熟悉 NV GPU 体系架构或海思 NPU 体系结构的优先； 5.熟悉高性能计算的优化方法，或有深boss度学习算子优化加速经验者优先；