大模型推理训练优化工程师
岗位摘要
PyTorch/Paddle模型适配与性能调优;多机多卡大模型训练部署与性能调优;vllm/SGLang以及推理引擎部署与性能调优;开发高性能算子;对社区CUDA仓库进行musify;调研最新的机器学习技术和平台功能
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- PyTorch/Paddle模型适配与性能调优
- 多机多卡大模型训练部署与性能调优
- vllm/SGLang以及推理引擎部署与性能调优
- 开发高性能算子
- 对社区CUDA仓库进行musify
- 调研最新的机器学习技术和平台功能
- 深度参与研发产品特性开发,为产品支持新特性
任职要求
- 积极乐观,责任心强,工作认真细致,具有良好的团队沟通与协作能力
- 计算机科学、电子工程或相关领域的本科及以上学历
- 熟悉机器学习和深度学习算法及其在实际应用中的部署
- 至少有一种机器学习框架使用或开发经验,如TensorFlow、PyTorch、PaddlePaddle、Megatron-LM、Colossal-ai、DeepSpeed、vLLM、SGLang或其他推理引擎
- 熟悉至少一种编程语言,如CUDA、C++、Triton、Python
- 良好的逻辑思维能力,精通计算机数据结构和算法
福利待遇
- 薪资待遇优厚,日薪200-300元
- 参与前沿AI大模型训练与推理优化项目
- 探索GPU性能极限,积累行业领先技术经验
- 团队氛围开放,沟通协作顺畅
工作地址
北京朝阳区望京国际研发园l座3层
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。