岗位职责
- 负责研发多模态生成、多模态理解等方向模型训练、推理核心算子的高性能优化工作；
- 支持抖音、剪映、即梦等产品生成式AI算法模型落地，支撑字节跳动AI技术的发展。
职位要求
- 本科及以上学历，计算机/电子/信息/通信/自动化/软件等相关专业优先；
- 有AI工程系统优化经验；深入理解计算机体系结构、熟悉CUDA/AscendC/BangC中的一种或者多种，熟悉计算密集型算子如Flash Attention、Conv2d、GEMM、GEMV实现和加速；
- 具有低比特量化、稀疏等优化技术经验者优先；
- 了解Torch.compile、TVM、MLIR等编译优化原理并具备相关技术优化经验优先；
- 了解主流AIGC、多模态算法模型原理，有AIGC模型并行训练或推理优化经验优先；
- 具备良好的交流沟通能力、业务意识、团队合作精神。
