岗位描述(节选)
1、参与视觉生成/多模态模型(包括文本、图像、视频生成等)在 GPU、ASIC、FPGA 等异构硬件上的推理/后训练加速开发与软硬件结合的性能优化工作,包括但不限于模型量化、attention优化、显存优化、编译优化、计算与通信优化、内存管理以及多卡或多设备的并行推理方案等;
2、在主流深度学习框架(如 PyTorch)基础上,基于GPU/xPU硬件特点,对关键算子进行软硬件结合优化,提升模型运行效率;
3、与硬件以及算法工程师紧密配合,共同优化整体推理速度与资源占用;
4、跟踪学术界与工业界前沿技术(如扩散模型优化、VAE并行优化、AI编解码、面向机器的编解码等),推动软硬件协同创新。
【任职要求】
1、计算机/电子/数学等相关专业,精通C++/Python/Rust,具备5年以上高性能计算或GPU相关软件开发经验。
2、熟悉常用数据结构与算法;对计算机体系结构(如缓存/多线程/SIM……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。