1. 參與AI算子開發(fā)與優(yōu)化，實現(xiàn)深度學習模型中的核心計算算子，針對不同硬件平臺（如GPU/NPU/CPU/DSP）進行性能優(yōu)化，提升計算效率；
2. 針對國產加速設備設計深度學習高性能算子方案，完成功能研發(fā)及適配優(yōu)化，保障硬件算力高效釋放；
3. 負責算子與主流AI框架（如PyTorch、TensorFlow、Caffe等）的集成，完成功能驗證、性能測試及穩(wěn)定性保障，確保算子穩(wěn)定運行；
4. 設計并開發(fā)算子自測試用例，搭建功能和性能分析工具，提升算子庫開發(fā)效率與質量；
5. 整理技術文檔、優(yōu)化記錄及開發(fā)文檔，協(xié)助團隊解決算子相關技術問題，支持模型部署、調優(yōu)及工程落地。

任職資格

1. 計算機、人工智能等相關領域碩士及以上學歷，具備扎實的C/C++/Python編程能力；
2. 良好的算法基礎，熟悉時間復雜度、空間復雜度分析，掌握軟件Debug技巧，能高效定位并解決問題；
3. 理解深度學習理論，熟悉PyTorch、TensorFlow等至少一種開源深度學習框架，了解框架底層算子運行機制；
4. 有CPU/GPU/NPU等硬件平臺底層函數(shù)庫開發(fā)或性能優(yōu)化經驗，熟悉圖優(yōu)化、算子融合等性能優(yōu)化方法者優(yōu)先；
5. 具備較強的快速學習能力、團隊合作能力及溝通能力，能適應跨團隊協(xié)作場景。

加分項

1. 有匯編優(yōu)化、CUDA、TensorRT、OpenCL、MPI并行計算或HPC開發(fā)經驗；
2. 有國產加速設備算子開發(fā)或適配經驗，或推理引擎（如TVM）實現(xiàn)原理相關經驗；
3. 有深度學習領域（目標檢測、機器視覺等）工程落地經驗；
4. 具備RDMA、DSP、FPGA等硬件編程或優(yōu)化經驗。