logo of horizon

地平线大模型推理优化(算子开发)

社招全职3年以上软件序列地点:北京 | 南京 | 上海状态:招聘

任职要求


1.计算机/电子工程硕士以上学历, 精通C++/CUDA,掌握Cutlass/Triton等高性能库开发
2.3年以上AI算子优化经验,主导过千亿参数模型优化项目
3.深入理解LLM架构(MHA/GQA/MO…
登录查看完整任职要求
微信扫码,1秒登录

工作职责


1.基于地平线BPU芯片架构设计开发高性能计算库,针对LLM核心算子(Attention/FFN/MOE等)进行硬件适配优化
2.与芯片设计团队协作,提出计算库需求并验证硬件性能瓶颈,推动芯片架构迭代
3.开发混合精度(FP8/INT4)算子库,适配地平线征程系列芯片的异构计算特性
包括英文材料
学历+
C+++
CUDA+
还有更多 •••