深度学习框架图编译工程师当面告知
中科寒武纪科技股份有限公司
2027-01-03 22:16:49
其他社招
岗位详情
地区：
北京市
专业要求：
人工智能计算机科学与技术软件工程
岗位职责：
1. 优化Torch Compile模式下各类神经网络的性能，包括网络性能瓶颈分析、编译耗时优化、codegen性能优化等；
2. 紧跟PyTorch社区最新技术进展，及时将前沿性能优化技术迁移到MLU平台，保持技术竞争力和创新性。
招聘要求：
1. 有较强的C++和Python功底，有扎实的数据结构和算法基础；
2. 学习能力强，有强烈的求知欲和自我驱动能力，良好的通能力和团队协作能力。
加分项1：熟悉PyTorch/TensorFlow等主流深度学习框架，熟悉框架内部的模块、机制。
加分项2：熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术。
公司介绍
中科寒武纪科技股份有限公司
中科寒武纪科技股份有限公司，成立于2016-03-15，位于北京，法定代表人：陈天石，登记状态：存续，注册资本41835.0224万人民币，公司类型：民营/私企
公司信息:
公司名称:中科寒武纪科技股份有限公司
公司类型:
公司性质:--
公司地点:
公司规模:
涉及领域:--
公司网址: