大模型训推优化实习岗(北杭深)当面告知
北京达佳互联信息技术有限公司
招满即止
社招(不限)博士实习硕士
岗位详情
地区:
全国
专业要求:
专业不限
岗位职责:
1、负责文生图、文生视频等自研大模型核心算子的自研开发与性能优化,基于CUDA、OpenAI Triton等工具实现高性能计算加速;
2、针对AI Infra大模型推理与训练场景,优化混合精度量化策略,设计低比特计算、稀疏化压缩等方案,提升模型部署效率;
3、深入GPU/Tensor Core硬件架构,优化显存管理、计算图调度及分布式通信,提升模型训练吞吐与资源利用率;
4、搭建端到端模型推理流水线,探索多模态生成任务下的算子融合、动态编译等创新优化手段;
5、协同算法团队完成模型轻量化落地,提供量化感知训练(QAT)、模型剪枝等技术支持。
招聘要求:
1、计算机科学、电子工程或相关专业硕士及以上学历,***CUDA/GPU高性能计算开发经验;
2、精通PyTorch/Sglang/vLLM等框架底层实现,熟悉大模型部署与推理优化原理;
3、熟练掌握OpenAI Triton编程,具备算子内核开发经验(如矩阵乘、Attention、Conv等模块优化);
4、熟悉模型量化技术(INT8/FP8/INT4混合精度)优先;
5、对GPU硬件架构(如Hopper/Ada)有深入理解,能结合硬件特性设计高性能算子。
加分项:
1、在CVPR/NeurIPS等顶会发表过模型压缩、加速相关论文或开源项目贡献者;
2、有文生视频、3D生成等长序列生成模型的并行化优化经验;
3、熟悉分布式训练框架(DeepSpeed/Megatron)或AI芯片(如NPU)开发经验;
4、具备多模态大模型(如Diffusion Model)端到端部署落地经验。
公司介绍
北
北京达佳互联信息技术有限公司
北京达佳互联信息技术有限公司,成立于2014-07-02,位于北京,法定代表人:杨远熙,登记状态:存续,注册资本600000万美元,公司类型:民营/私企
公司信息:公司名称:北京达佳互联信息技术有限公司公司类型:媒体/出版/影视/文化传播公司性质:5000以上公司地点:公司规模:--涉及领域:--公司网址: