职位详情
微信扫码分享
投诉
面向公司自研AI芯片,承担基于Triton的神经网络算子设计与性能调优工作;
职位要求:
1、本科及以上学历,具备扎实的计算机基础知识,熟练掌握C/C++及Python语言;
2、具备一定的算子开发背景(如卷积、注意力机制、矩阵乘法等);
3、了解编译原理、中间表示、后端实现及相关优化技术;
4、熟悉计算机体系结构原理;
5、对深度学习编译器有深入理解,掌握TVM、XLA、MLIR或Halide中至少一项者优先;
6、熟悉主流深度学习算法及框架(包括caffe、tensorflow、pytorch、mxnet等);
7、具备在nvidia、AMD、Intel、Qualcomm、arm、x86等架构之一进行性能优化的经验,国产芯片经验亦可接受;
8、曾主导或参与开源软件开发,或为知名开源项目贡献代码者优先(尤其AI编译器方向)。

杨女士IP:浙江杭州
2周内活跃|
中昊芯英(杭州)科技有限公司
相关推荐查看更多 >
杭州
相同职位推荐
企业招聘

