算子优化工程师——创新业务
Technology
北京曦望芯科智能科技有限公司上海-浦东新区, 中国¥50,000 - ¥80,000 /月1个月前
职位描述
所属部门:创新业务部门 职位描述 1、设计,实现优化算子库代码,如flash-attention/page-attention等; 2、定义并实现新一代芯片的AI算子算法代码及性能调优; 3、研究前沿的AI算法,实现并进行性能分析,提供下一代芯片改进方案。 职位要求 1、熟悉linux开发环境,python,C语言; 2、熟悉GPU或者AI芯片的硬件架构; 3、熟悉常用AI算子,如flash-attention; 4、有手写算子或者汇编优化经验优先。
Keywords
5-10年Python
对这个职位感兴趣吗?