锦鲤求职

江原科技

AI软件架构师(异构计算和底层通信方向)

锦鲤会替你打开官网网申、按简历自动填表提交,你只需要在关键步骤确认。

岗位描述

岗位职责
1. HCC平台总体架构设计:定义HCC软件平台的逻辑架构,明确编程模型(用户态API)、集合通信库、运行时、驱动、工具链的模块边界与交互协议。制定与硬件团队协同的软硬件接口规范(如指令集、寄存器布局、内存模型、中断语义、互联拓扑配置)。
2. 编程模型与运行时架构:设计面向异构场景的统一编程模型(类似CUDA),包括内核语言扩展、主机端API、流/事件模型、内存管理(UVA、分页锁定内存等)。制定运行时系统架构(任务提交、同步、错误处理、设备热插拔),确保低延迟和高吞吐。
3. 集合通信库架构方案:设计HCC通信库的整体架构(通信原语集、拓扑发现、算法自动选择、通信-计算重叠机制)。针对江原芯片的片间互联拓扑(如环形、全互联、Mesh),制定拓扑感知的通信算法选型方案(如Ring vs Tree vs Halving-Doubling)。设计多节点扩展方案(通过RoCE/InfiniBand或自定义协议),支持跨主机集群通信。
4. 驱动与内核架构:制定内核态驱动架构(中断处理、命令队列、DMA引擎、IOMMU映射、虚拟化支持)。设计与用户态驱动/运行时的接口规范,确保安全性与性能的平衡。
5. 工具链架构方案:设计Profiler的架构(软件打点、硬件计数器读取、数据聚合与可视化方案),支持算子级、通信流级的性能分析。制定汇编器/反汇编器、内存检查器、仿真器等工具的集成方案。
6. 架构落地与推进:输出HCC平台架构文档、接口规格和迭代Roadmap,组织跨团队(硬件、驱动、应用软件)技术评审。推动方案在硅前仿真环境、FPGA原型、芯片流片后的分阶段落地,识别并解决集成风险(如驱动与硬件时序不匹配、通信库死锁)。

任职要求
必选项:计算机、电子工程等相关专业硕士及以上学历,8年以上底层系统软件经验,其中至少3年担任架构师或技术Leader角色。具备从0到1设计异构计算/通信软件架构并落地量产的经验(不限NPU,GPU/DPU/智能网卡亦可)。
编程模型:深入理解至少一种异构编程模型(CUDA/ROCm/OpenCL)的完整设计,主导过运行时或设备驱动层的架构方案。
集合通信:精通集合通信库(NCCL/RCCL/MPI)的架构设计,有主导通信算法优化(如Ring AllReduce、Double Tree)的落地案例。
驱动与内核:熟悉Linux内核驱动架构(PCIe子系统、DMA、中断、IOMMU),主导过高性能设备驱动的方案设计与发布。
软硬件协同:能阅读硬件规格书(寄存器描述、指令集、互联协议),并与硬件工程师讨论软硬件接口划分方案。
优秀的系统抽象和接口定义能力,能设计清晰、可演进的API。
具备跨团队协同能力(驱动团队、硬件团队、应用团队),推动架构方案对齐并落地。

其他
参与过定制互联协议(如类NVLink、CCIX、CXL)的主机端软件栈设计。
有RDMA、RoCEv2、InfiniBand编程经验或驱动开发经验。
熟悉SR-IOV、VFIO、设备虚拟化方案设计。

高端技术职位方向的申请准备

先核对岗位要求与自己的经历,再准备档案、投递和面试。

阅读求职步骤与示例 →