岗位描述
岗位职责
1、从事AI Runtime/UMD软件的研发工作;
2、参与面向深度学习AI芯片的runtime/UMD软件栈的设计、开发与日常维护;
3、负责协同编译器和内核态驱动团队,在不同平台下适配驱动和Runtime并做性能优化。
岗位要求
1、计算机软件类/电子信息类/集成电路/人工智能类等相关专业,精通C++编程语言;
2、熟悉类CUDA/ROCm Runtime和Driver API及其底层实现,理解常用机制如异步launch、异步DMA、多stream、事件event,有进程隔离/调度,MIG开发经验;
3、熟悉CUDA runtime, AMD ROCm/clr 等软件,对CUDA兼容runtime/UMD有过开发经验者优先;
4、熟悉至少一种异构计算编程体系,如:CUDA、HIP、SYCL、OpenCL等;
5、有很好的团队协作能力与沟通能力,对技术和代码品质有追求; (加分项)具有神经网络开发经验,熟悉PyTorch和TensorFlow等深度学习框架; (加分项)在模型训练、量化和部署方面有经验,尤其是在大型语言模型(LLM)方面。