岗位职责
- 基于自研芯片开发Runtime/UMD/KMD软件栈;
- 参与芯片软件栈的架构设计,包括任务调度、内存管理、计算资源分配等核心子系统的实现与优化;
- 参与硬件整个设计流程,迭代驱动设计以满足model/emulation/hardware的需求;
- 分析和解决KMD/UMD的功能、性能和稳定性问题,确保高质量软件交付;
- 构建性能分析、调试与 profiling 工具链,为内外部开发者提供性能洞察与优化建议;
- 跟踪行业趋势和实践,持续改进driver的设计和实现;
职位要求
- 计算机相关专业,本科及以上学历,5年以上底软开发经验,熟练掌握C/C++/Makefile/bash等语言的使用;
- 对GPGPU/NPU或类似大规模并行处理器(如NVIDIA CUDA, AMD ROCm)的软硬件架构有深刻理解,具备丰富的用户态驱动或高性能库开发经验;
- 精通Linux内核,包括但不限于bootloader,init,内存管理,设备驱动等;
- 精通C/C++,具有异构Runtime开发和量产交付经验优先;
- 熟悉TVM/MLR等框架Runtime系统,熟悉AI模型推理部署链路,内存管理等;
- 熟悉深度学习算法、大模型基本结构,熟悉相关推理框架,如llama.cpp、vLLM、OLLAMA 等;