锦鲤求职岗位描述
岗位职责
参与自研AI芯片对PyTorch等主流AI框架的适配与深度优化，确保大模型在芯片上高效运行。
负责或协助开发PyTorch的第三方Device后端插件（类似CUDAPlugin），实现框架与芯片Runtime的无缝对接。
优化框架层面的内存管理、算子分发策略，以及分布式训练的通信封装，提升框架在异构硬件上的易用性和性能。
跟踪PyTorch社区最新特性，并将其适配到自研芯片平台。
任职要求
核心技能：熟练掌握Python和C++，有丰富的PyTorch/TensorFlow等框架的使用和源码阅读经验。
其他
加分项（必备其一）：了解PyTorch的内部实现机制（如Autograd、JIT）；有为PyTorch添加Custom Device/Operator的经验；参与过开源AI框架社区贡献。
硕士在读（2026届优先），能保证每周4天以上、至少6个月的实习。
你将收获：掌握AI框架与底层硬件结合的桥梁技术，成为连接上层算法与下层芯片的关键角色。
在框架开发专家的指导下，系统学习大型开源项目的工程化协作与开发流程。
表现优异者（最快3个月）可获校招转正Offer，加入推动国产AI框架生态繁荣的行列。