小米顶尖应届-AI Compiler 开发工程师-芯片
任职要求
1、计算体系结构或者编译技术方向博士;2、从事GPGPU/NPU上AI编译技术研究,并且发表过顶级论文;3、有较强的分析和解决问题能力、较强的动手能力;4、熟悉L…
工作职责
1、结合端侧NPU和模型深入研究编译技术,例如针对计算单元的自动向量化,深度融合等; 2、研究优化针对Tensor Core计算切分,多种并发模式优化等; 3、研究访存优化,Layout最优化求解等; 4、利用AutoTuning、机器学习等技术寻找最优化配置方案等; 5、探索端到端模型编译新方法,Tile-based DSL设计与开发等。
1. 深入理解自研芯片架构、指令集及性能瓶颈,开展面向业务场景的芯片亲和性算法设计与优化 。 2. 结合人车家等业务场景研究并实现前沿的模型优化、量化等技术 ,并针对目标芯片进行定制化适配与验证,并形成指导下一代芯片设计的算法输入 。 3. 探索并应用AI Agent等新范式开发与优化面向自研芯片的高性能AI Infra框架、编译器、算子库及运行时系统等系统软件。 4. 参与构建从算法模型到芯片部署的完整工具链、性能分析平台与自动化优化流程 。
1. 下一代AI编译器架构设计与研发:主导AI芯片编译器的核心方案设计与开发实现,聚焦于软硬件协同优化,挑战极致的硬件资源利用率与高能效比(Energy Efficiency)。 2. LLM驱动的编译器前沿探索:结合大语言模型(LLM)与智能体(Agent)技术,探索自动代码生成、智能算子融合及自动化性能调优等前沿课题,构建“AI辅助AI编译”的下一代智能工具链。 3. 软硬协同的系统级优化:深度融合前沿AI算法(如大模型推理/训练)、底层硬件架构与业务产品需求,打通从算法到芯片的端到端编译优化链路。 4. 极致性能与功耗分析:主导复杂模型在真实场景下的性能瓶颈剖析,建立深度的性能与功耗分析模型,研发并落地系统级的极致优化策略。 5. 编译器技术演进与生态建设:保持对业界顶尖编译器(如TVM, MLIR, Triton等)及LLM前沿动态的敏锐嗅觉,规划并引领自研AI编译器的技术演进路线图。
1、架构方案前瞻探索:围绕后续AI芯片产品,开展面向模型领域的架构方案探索与选型,为产品技术路线图提供决策依据,直接影响芯片竞争力; 2、算法-架构协同优化:针对量化、稀疏、注意力结构等算法演进趋势,建立算法参数与硬件架构的适配映射模型,确保芯片架构能够高效支撑算法快速迭代; 3、存储层级方案设计:结合HBM、3D DRAM、HBF等存储介质特性,探索面向不同推理场景的最优存储架构方案,直接影响推理性能与成本竞争力; 4、设计空间探索与评估:构建系统级架构评估方法,在算法与硬件的联合空间中高效搜索最优方案,降低架构决策风险,缩短研发周期; 5、前沿技术预研储备:持续跟踪AI芯片架构领域前沿方向,沉淀核心技术Know-how与人才储备,为团队长期技术竞争力奠定基础。