XPENG

AI编译器&模型优化实习生

XPENG  •  Onsite  •  2 months ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

AI编译器&模型优化实习生北京、上海实习研发 - 算法职位描述1、参与面向大模型与物理AI场景的AI编译器研发,优化计算图表示、算子融合与内存调度策略;
2、针对自研AI芯片进行算子定制与性能调优,实现端到端推理与训练加速;
3、设计并实现自动代码生成工具链,支持多后端(GPU/NPU/CPU)的高效算子发射;
4、调研SOTA大语言模型的压缩和加速算法,并针对小鹏的模型结构做优化和实现;
5、与算法、芯片团队深度协作,推动编译优化在百万级量产环境中的稳定落地。职位要求1、27届-28届毕业的同学,计算机/电子工程等相关专业,硕士及以上在读;
2、扎实的系统与工程能力:精通C++/Python,扎实掌握数据结构与算法,能高质量实现高性能算子或底层系统组件;
3、AI编译器基础:熟悉MLIR/TVM/XLA/IREE等编译框架,了解计算图优化、算子融合、内存复用等关键技术;
4、模型理解:熟悉LLM/VLM/扩散模型等主流架构,理解其计算模式对编译优化的需求;熟悉常用的大语言模型优化和压缩算法,比如量化,剪枝,稀疏化。
【加分项】
1、软硬协同优化:深度参与过AI芯片、编译器或软硬协同优化,对算力调度与成本优化有实践积累;
2、性能极致追求:有AutoML、模型压缩、量化或推理加速(如PagedAttention、Contiguous KV Cache)经验,或在Kaggle/天池等竞赛中取得优异成绩;
3、学术影响力:在OSDI/SOSP/ISCA/MLSys/PLDI等顶会发表过编译器/系统相关论文,或有高质量GitHub开源项目/工具贡献(如TVM/MLIR社区贡献)。 投递
XPENG

About XPENG

XPeng is a leading Chinese Smart EV company that designs, develops, manufactures, and markets Smart EVs that appeal to the large and growing base of technology-savvy middle-class consumers. Its mission is to drive Smart EV transformation with technology and data, shaping the mobility experience of the future. In order to optimize its customers’ mobility experience, XPeng develops in-house its full-stack advanced driver-assistance system technology and in-car intelligent operating system, as well as core vehicle systems including powertrain and the electrical/electronic architecture. XPeng is headquartered in Guangzhou, China. In 2021, the Company established its European headquarters in Amsterdam, along with other dedicated offices in Copenhagen, Munich, Oslo, and Stockholm.The Company’s Smart EVs are mainly manufactured at its plant in Zhaoqing and Guangzhou,Guangdong province.

For more information, please visit https://heyxpeng.com.

Industry
Automotive & Mobility
Company Size
1,001-5,000 employees
Headquarters
Guangzhou, CN
Year Founded
2014
Social Media