北极雄芯信息科技（西安）有限公司

【岗位职责】 岗位职责： 1、围绕公司自研 3D-NPU 芯片，参与 AI 推理基础设施的设计与开发，包括 PyTorch Backend、算子库、运行时及推理框架适配等； 2、负责大模型推理系统的工程化落地与性能优化，涵盖模型加载、显存管理、计算调度、算子融合、连续批处理和 KV Cache 管理等关键模块； 3、参与多卡、多机及超节点推理能力建设，开发和优化设备通信、集合通信、任务调度及分布式部署等基础组件； 4、打通硬件、编译器、运行时/驱动和框架全链路，定位计算、访存、通信及资源调度等方面的性能瓶颈，推动问题闭环和产品性能持续提升。； 5、建设 AI Infra 相关工程平台与工具链，包括自动化测试、性能评测、故障诊断、监控分析及持续集成体系； 6、参与推理框架及基础软件的架构设计、代码开发和质量建设，保障系统的高性能、稳定性、可扩展性与易用性。 【任职要求】 岗位职责： 1、围绕公司自研 3D-NPU 芯片，参与 AI 推理基础设施的设计与开发，包括 PyTorch Backend、算子库、运行时及推理框架适配等； 2、负责大模型推理系统的工程化落地与性能优化，涵盖模型加载、显存管理、计算调度、算子融合、连续批处理和 KV Cache 管理等关键模块； 3、参与多卡、多机及超节点推理能力建设，开发和优化设备通信、集合通信、任务调度及分布式部署等基础组件； 4、打通硬件、编译器、运行时/驱动和框架全链路，定位计算、访存、通信及资源调度等方面的性能瓶颈，推动问题闭环和产品性能持续提升。； 5、建设 AI Infra 相关工程平台与工具链，包括自动化测试、性能评测、故障诊断、监控分析及持续集成体系； 6、参与推理框架及基础软件的架构设计、代码开发和质量建设，保障系统的高性能、稳定性、可扩展性与易用性。 【福利待遇】 项目奖励，绩效奖金，中长期奖励，年终奖 六险一金，通讯补贴，加班补贴，下午茶，带薪年假，健康体检 1.网站跳转投递： https://app.mokahr.com/campus-recruitment/bjxx/172674?locale=zh-CN 【职业发展】 北极雄芯源于清华大学交叉信息研究院，孵化于交叉信息核心技术研究院。公司成立于2021年7月，是国家高新技术企业，陕西省"专精特新"中小企业。在西安、北京、南京、无锡等地设立了研发中心。作为国内领先的Chiplet芯片设计企业，公司核心团队深耕Chiplet领域多年，拥有全球领先的学术影响力以及丰富的产品化实践基础。公司致力于成为基于Chiplet的高性能计算解决方案领航者，成立以来取得了多项里程碑。公司聚焦大模型时代的端云算力需求，致力于以高效能算力解决方案，支撑大模型在终端与云端的规模化部署，自主研发的 2D 与 3D Chiplet（芯粒）产品已陆续进入量产阶段，为高算力、低功耗的芯片系统提供了可落地的技术路径。 【其他信息】 职位类别：深度学习工程师 招聘人数：5人 专业要求：电子科学与技术 / 微电子学与固体电子学 / 计算机科学与技术 / 电子科学与技术 / 电子与通信工程 / 计算机技术 / 软件工程 联系方式：朱女士 029-81292600 公司地址：陕西省西安市高新区天谷八路156号西安软件新城研发基地2期B2座6层