logo of xpeng

小鹏汽车AI编译器&模型优化实习生

实习兼职地点:北京 | 上海状态:招聘

任职要求


1、27届-28届毕业的同学,计算机/电子工程等相关专业,硕士及以上在读;
2、扎实的系统与工程能力:精通C++/Python,扎实掌握数据结构算法,能高质量实现高性能算子或底层系统组件;
3、AI编译器基础:熟悉MLIR/TVM/XLA/IREE等编译框架,了解计算图优化、算子融合、内存复用等关键技术;
4、模型理解:熟悉LLM/VLM/扩散模型等主流架构,理解其计算模式对编译优化的需求;熟悉常用的大语言模型优化和压缩算法,比如量化,剪枝,稀疏化。
【…
登录查看完整任职要求
微信扫码,1秒登录

工作职责


1、参与面向大模型与物理AI场景的AI编译器研发,优化计算图表示、算子融合与内存调度策略;
2、针对自研AI芯片进行算子定制与性能调优,实现端到端推理与训练加速;
3、设计并实现自动代码生成工具链,支持多后端(GPU/NPU/CPU)的高效算子发射;
4、调研SOTA大语言模型的压缩和加速算法,并针对小鹏的模型结构做优化和实现;
5、与算法、芯片团队深度协作,推动编译优化在百万级量产环境中的稳定落地。
包括英文材料
C+++
Python+
数据结构+
算法+
还有更多 •••
相关职位

logo of liauto
社招5年以上智能与信息技术

1、与芯片架构师合作,协助芯片架构设计,参与制定芯片软硬件协作方案; 2、参与AI算子库的开发和优化,支撑算法所需算子的功能和基本性能要求; 3、参与AI编译器的设计研发,由AI编译器将算法模型和算子库编译成AI芯片的可执行文件; 4、维护AI编译器和算子库,解决在Virtual platform、RTL simulation、FPGA验证中发现软件功能或性能问题。

上海
logo of liauto
实习算法与软件

1、承接端到端自动驾驶/大语言类AI模型负载,为大算力芯片研发设计AI模型调度、编译软件栈,实现高性能推理或训练。 2、参与数据流模式时空调度建模与算法开发,支撑AI模型自动化调度到大算力芯片之上,达到较高端到端性能。 3、参与AI基本算子的开发和优化,支撑算法模型推理所需算子的功能和基本性能要求,分析性能瓶颈,构建方案极致优化。

北京
logo of xpeng
社招

主要负责ai模型量化和编译器相关的开发, 工作内容包括, 1.量化工具及模型量化,包括QAT和PTQ等相关的各种工具和算法 2.计算图优化及模型部署,包括各种网络的常见通用优化

更新于 2024-09-05上海
logo of xpeng
社招

主要负责AI编译器相关的开发和交付, 工作内容包括, 1. 各种网络的常见优化,包括内存调度,切分,并行等 2. 模型部署以及性能调试 3. clang/LLVM编译器工具链相关功能开发

更新于 2024-09-05上海