资深大模型训练推理算子高性能优化工程师-Data AML

关于这个机会

1、负责研发多模态生成、多模态理解等方向模型训练、推理核心算子的高性能优化工作； 2、支持抖音、剪映、即梦等产品生成式AI算法模型落地，支撑字节跳动AI技术的发展。

任职要求

1、本科及以上学历，计算机/电子/信息/通信/自动化/软件等相关专业优先； 2、有AI工程系统优化经验；深入理解计算机体系结构、熟悉CUDA/AscendC/BangC中的一种或者多种，熟悉计算密集型算子如Flash Attention、Conv2d、GEMM、GEMV实现和加速； 3、具有低比特量化、稀疏等优化技术经验者优先； 4、了解Torch.compile、TVM、MLIR等编译优化原理并具备相关技术优化经验优先； 5、了解主流AIGC、多模态算法模型原理，有AIGC模型并行训练或推理优化经验优先； 6、具备良好的交流沟通能力、业务意识、团队合作精神。

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。