
MomentaAI编译器专家
任职要求
任职要求: 1.计算机科学、电子信息工程或相关专业硕士学位以上,3年以上相关工作经验。 2.精通C/C++和Python等程序设计语言,具备扎实的编码能力。 3.熟悉TensorFlow、PyTorch、ONNX等深度学习框架的使用和内部机制。 4.深入理解编译器架构和原理,具备编译器或嵌入式系统开发经验。 5.拥有GPU、NPU、DSP等异构计算平台的开发和性能优化经验。 6.熟悉自驾场景者优先。
工作职责
岗位职责: 1.负责设计、开发和维护DSA架构下的NPU编译器工具链,确保其高效稳定运行。 2.扩展和优化深度学习框架(包括TensorFlow、PyTorch、ONNX等)的支持能力,提升AI模型兼容性。 3.负责计算图优化,包括各种网络的常见通用优化以及针对硬件平台的优化。 4.优化编译器工具链中的各种算法,以提高编译质量和执行效率。 5.与芯片设计团队及软件开发团队紧密合作,进行系统层面的编译器性能调优。 6.对开源编译器框架TVM/LLVM(MLIR)进行二次开发,以满足自驾业务需求。
作为 AI 加速和 AI 编译器专家,负责开发和优化AI算法的编译器和工具链,设计和实施软硬件协同策略,推动AI技术在多个平台上的高效执行和应用。 1. 设计、开发和优化面向云上AI应用的Compiler Toolchain,提高算法在硬件上的性能和效率。 2. 研究和实现先进的软硬件协同设计方法,以优化AI模型的运行时性能和能效。 3. 与算法研发团队合作,确保AI模型的高效部署和执行。 4. 持续跟踪行业最新动态,评估和集成新的编程模型和软硬件技术。
概括 想象一下您在这里能做什么! 在小米,新想法可以很快变成非凡的产品、服务和客户体验。 对工作充满真诚热爱,您将取得的成就前所未有。 充满活力、勤奋的员工和鼓舞人心的创新技术是这里的常态。 在这里工作的人通过小米的硬件和软件产品影响整个行业。 加入我们,帮助推出下一个突破性的小米产品。 本团队正在开展小米AI领域的硬件/软件协同设计创新,以实现高效AI推理 我们正在寻找一位积极主动的研究工程师加入我们的团队,他/她应具有强大的AI系统背景和软件开发实践经验。 该职位的理想人选是一位研究人员和工程师,他/她将不断突破现有界限,并以影响小米内部外部合作伙伴和更广泛的研究社区来实现设想。 描述: 您将主要专注于构建端侧AI模型、编译器、推理框架的优化,以便在所有小米手机上高效的AI推理 开发高效的调试工具链,以便找到从模型本身到AI芯片或者CPU运算单元每一个环节的瓶颈点 与业界最佳成果进行对标,挖掘优化点,识别优先级,与相关团队配合论证可行性,并且最终落地优化成果 总结你的优化经验,整合小米从软件到芯片的能力,让开发者可以迭代模型创作、优化、转换、执行、调试、分析
概括 想象一下您在这里能做什么! 在小米,新想法可以很快变成非凡的产品、服务和客户体验。 对工作充满真诚热爱,您将取得的成就前所未有。 充满活力、勤奋的员工和鼓舞人心的创新技术是这里的常态。 在这里工作的人通过小米的硬件和软件产品影响整个行业。 加入我们,帮助推出下一个突破性的小米产品。 本团队正在开展小米AI领域的硬件/软件协同设计创新,以实现高效AI推理 我们正在寻找一位积极主动的研究工程师加入我们的团队,他/她应具有强大的AI系统背景和软件开发实践经验。 该职位的理想人选是一位研究人员和工程师,他/她将不断突破现有界限,并以影响小米内部外部合作伙伴和更广泛的研究社区来实现设想。 描述: 您将主要专注于构建端侧AI模型、编译器、推理框架的优化,以便在所有小米手机上高效的AI推理 开发高效的调试工具链,以便找到从模型本身到AI芯片或者CPU运算单元每一个环节的瓶颈点 与业界最佳成果进行对标,挖掘优化点,识别优先级,与相关团队配合论证可行性,并且最终落地优化成果 总结你的优化经验,整合小米从软件到芯片的能力,让开发者可以迭代模型创作、优化、转换、执行、调试、分析