蚂蚁集团-运行时Runtime工程师-北京/上海

岗位职责

1. 基于自研芯片开发Runtime/UMD/KMD软件栈；
2. 参与芯片软件栈的架构设计，包括任务调度、内存管理、计算资源分配等核心子系统的实现与优化；
3. 参与硬件整个设计流程，迭代驱动设计以满足model/emulation/hardware的需求；
4. 分析和解决KMD/UMD的功能、性能和稳定性问题，确保高质量软件交付；
5. 构建性能分析、调试与 profiling 工具链，为内外部开发者提供性能洞察与优化建议；
6. 跟踪行业趋势和实践，持续改进driver的设计和实现；

任职要求

1. 计算机相关专业，本科及以上学历，5年以上底软开发经验，熟练掌握C/C++/Makefile/bash等语言的使用；
2. 对GPGPU/NPU或类似大规模并行处理器（如NVIDIA CUDA, AMD ROCm）的软硬件架构有深刻理解，具备丰富的用户态驱动或高性能库开发经验；
3. 精通Linux内核，包括但不限于bootloader，init，内存管理，设备驱动等；
4. 精通C/C++，具有异构Runtime开发和量产交付经验优先；
5. 熟悉TVM/MLR等框架Runtime系统，熟悉AI模型推理部署链路，内存管理等；
6. 熟悉深度学习算法、大模型基本结构，熟悉相关推理框架，如llama.cpp、vLLM、OLLAMA 等；

岗位标签

NEW

特色标签

C、C++、Linux开发/部署经验、人工智能芯片、任务分配、内存优化、内存分配、内存调度、内核态驱动、内核模式驱动、嵌入式软件经验、性能分析、性能剖析、智能芯片、架构设计经验、用户态驱动、用户模式驱动、神经处理单元、神经网络处理器、算力分配、编译器开发经验、计算资源调度、调度管理、运行时环境、运行时系统、通用GPU、通用图形处理器