字节跳动 社会招聘
多模态大模型推理算子优化工程师-Data AML(北京/上海/杭州/深圳)
北京 ·经验不限·学历不限
北京
薪资面议
前往 字节跳动 官网投递
职位描述
1、负责Seedance/Seedream基础模型及衍生模型异构卡推理加速工作;
2、负责视频/图像生成/多模态等方向模型的异构卡适配。
【任职要求】
1、有AI工程系统优化经验,深入理解计算机体系结构、熟悉CUDA/AscendC/BangC/HIP中的一种或者多种算子,熟悉计算密集型算子如FlashAttention、Conv2d、Matmul、GroupedMatmul、MOE实现和加速;
2、具备低比特量化、稀疏等优化技术经验者优先;
3、熟悉Triton、TileLang、Cute DSL等语言并具备相关技术优化经验优先;
4、了解主流AIGC、多模态算法模型原理,有AIGC模型训练适配&加速、通算融合/掩盖经验优先;
5、熟悉TensorRT-LLM、SGLang、vLLM引擎,或熟悉ZeRO、TP、SP等并行技术原理优先;
6、具备良好的沟通能力和团队合作精神。
官网发布:2024-03-18 · 最后确认在招:2026-09-03 19:35:21 · 来源平台:feishu