logo of cambricon

寒武纪PyTorch框架研发实习生-27届

实习兼职地点:上海 | 北京状态:招聘

工作描述


岗位职责:
1. 负责PyTorch框架在寒武纪芯片上适配、优化与迭代;
2. 参与PyTorch社区开发, 跟踪社区最新进展, 完善和改进Cambricon Torch;
3. 优化PyTorch框架在寒武纪软件上运行速度, 提高模型训练和推理效率;
4. 开发PyTorch框架相关配套工具, 协助客户…
登录查看完整工作描述
微信扫码,1秒登录

包括英文材料
PyTorch+
性能调优+
学历+
C+
C+++
还有更多 •••
相关职位

logo of tencent
社招1年以上大数据(数据计算

1.熟练掌握pytorch框架,可对训练和推理任务进行性能分析和优化,具备框架特性开发的能力; 2.熟悉torch.compile(TorchDynamo、AOTAutograd、PrimTorch

更新于 2026-07-28深圳