返回信息流实习生主要参与开源项目:https://github.com/siliconflow/onediff
联系方式:https://www.shixiseng.com/intern/inn_o2c63agwogc7
硅动科技(SiliconFlow)是一家专注于大模型 Infra 技术创新的企业,已经推出用于生图生视频加速工具 OneDiff、AI 云平台 SiliconCloud、ComfyUI 开源节点 BizyAir。公司团队曾研发有较强技术影响力的分布式深度学习框架 OneFlow。
【岗位描述】
AIGC推理优化工程师(全职、实习)
【职位描述】
1、对视觉生成模型(如Stable Diffusion、DiT系列模型)做加速、部署;
2、开发加速基础设施 onediff,主要是开发高性能算子、分布式、编译工具;
【职位要求】
- 算子优化方向,了解 PyTorch 的算子开发流程,使用 NVIDIA Nsight/torch profiler 等分析性能,使用 CUDA、OpenAI Triton、CUTLASS 实现高性能算子,如实现融合算子、FlashAttention算子;
- 分布式方向,了解如何利用分布式通信支持模型的多机多卡并行执行,比如用数据并行、模型并行、流水并行以及大模型领域中的分布式技术加速模型;
- 熟悉Python、C++编程语言特性,熟悉 PyTorch,熟悉Linux/Git程序开发流程及环境;
- 积极主动,能与团队融洽沟通、合作;
- 本科及以上学历,计算机相关专业;
实习需要了解其中某个方向的某一两个点,至少实习3个月、每周4天.
【工作地点】
北京-搜狐网络大厦。
投递要求:
简历要求: 中文
工作地点:
北京市/北京市/海淀区 搜狐网络大厦
这是一条镜像帖。来源:北邮人论坛 / job-info / #967606同步于 2024/7/31
JobInfo机器人发帖
【校招】SiliconFlow视觉大模型推理加速实习生
threbody
2024/7/31镜像同步0 回复
订阅后,新回复会通过你的通知中心匿名送达。
0 条回复
暂无回复 · 你可以订阅本帖等待新回复。