logo of horizon

地平线【专项人才计划】编译优化-Agent算法实习生

校招全职软件序列地点:北京 | 上海状态:招聘

任职要求


教育背景:计算机,自动化,AI算法等相关专业在读硕士及以上学历。
AI与模型基础: 熟悉 PyTorchTensorFlow深度学习框架的底层机制,了解大语言模型的工作原理和 Prompt 调优方法。
编程与算子基础: 精通 Python,具备良好的 C++ 编程基础;对计算机内存层级优化…
登录查看完整任职要求
微信扫码,1秒登录

工作职责


Agent 优化策略验证: 参与探索基于大模型 Agent 的编译优化闭环系统,尝试利用 LLM 进行自动化的代码分析、性能瓶颈定位和调优策略生成(Auto-Tuning)。
前沿模型结构分析: 深入分析当前主流及前沿的深度学习模型结构(特别是各类 Transformer 变体,以及包含 FlashAttention、Linear Attention 等特性的模型),提取其计算和访存特征,为底层算子优化提供指导。
高性能算子开发: 结合理论分析与 LLM 代码生成能力,参与编写和优化针对公司 AI 芯片的高性能算子,挑战极致的执行效率。
工具链生态沉淀: 协助开发基于 LLM 的内部辅助工具,优化编译团队的研发工作流。
包括英文材料
算法+
学历+
PyTorch+
TensorFlow+
深度学习+
Prompt+
Python+
还有更多 •••