职位描述
工作职责:
1、参与GPU芯片的智算软件栈开发、优化和文档编写,包括但不限于GPU驱动、计算加速库2、与芯片架构、工具链团队深度协作,参与硬件关键功能、精度和性能评估3、协助其他团队推动芯片的验证、测试、调优工作4、针对特定场景进行性能瓶颈分析,给出切实可行的优化方向或方案
任职资格:
1、熟悉计算机体系结构,深入理解GPU架构(如SM单元、显存层次、线程束等);
2、扎实的C/C 编程与调试能力,良好的编程和代码管理习惯;
3、具有CUDA/ROCm/OpenCL开发经验,熟悉Pytorch、TensorRT等框架软件;
4、具备量产GPU芯片的软件开发经验,有GPU工程化程序优化者优先;
5、良好的自驱能力,具有较强烈的探索新技术领域的热情、快速学习新技术和解决复杂问题的能力。