职位详情
生日福利
团建聚餐
带薪年假
年终奖
定期体检
五险一金
昆仑万维-天工大模型
【昆仑万维公司介绍】:昆仑万维是中国领先的互联网平台出海企业,业务覆盖包括 AIGC 、信息分发、社交、 娱乐、元宇宙、游戏等多个领域,旗下包括 AGI 与 AIGC 、海外信息分发与元宇宙、投资等三大业务板块,市场遍及五大洲,70个国家和地区。全球平均月活跃用户近 4 亿, 2022年实现营业收入47.4亿元。
昆仑万维于2015年登陆创业板。
工作职责:
1、负责分布式高性能AI计算框架研发;
2、高性能 CUDA 算子研发
3、分布式训练调优和推理优化
4、网络通信优化和内存优化
5、MoE 训练优化
任职要求:
1、硕士及以上学历,计算机相关专业,较强的编程能力
2、精通 Python/C++,熟悉深度学习框架 PyTorch 及其底层实现
3、精通 计算机体系结构, 熟悉 GPU 架构 和 CUDA 编程
4、熟悉 Megatron-LM 、 DeepSpeed 等分布式训练框架
5、了解深度学习算法,包括 LLM、 CV、RL 等
6、有深度学习推理/训练系统开发经验者优先