独角兽

性能软件开发工程师/架构师

壁仞科技

上海人工智能社会招聘

岗位描述

岗位描述:1. 研究GPGPU及其集群中的性能场景,分析并定位软硬件性能瓶颈。2. 跟踪AI/HPC应用场景中的性能分析技术趋势,负责性能软件产品架构设计。3. 开发相关Benchmark、Profiler、性能模型、SDK等性能软件栈。4. 联合硬件架构师、框架团队、库函数团队等反馈性能问题,推动产品性能迭代。 岗位要求:1. 计算机科学、电子工程相关背景,深入理解计算机体系结构,思维敏捷,工程实践能力强。2. 具备扎实的C/C++/Python编程能力,熟悉CUDA编程模型,精通中间件与SDK开发。3. 精通Linux操作系统,或对GPU性能分析与优化有深入理解。4. 熟悉PyTorch、数学库(如DNN、BLAS等)、集合通信库(如CCL、SHMEM)以及分布式架构和调优方法。5. 熟悉数据挖掘、分析、建模与可视化方法,并具备动手开发能力。以上条件满足2项以上即可。