深度学习框架工程师
20-40K·14薪深圳市本科1-3年
职位描述
工作内容:负责公司自研芯片和开源三方框架集成;岗位职责:
1、框架调研,输出对应的介绍和需求文档;
2、测试和对比竞品软件栈;
3、负责自研芯片仿真库接入开源三方框架,分析性能瓶颈;
任职要求:
1、熟悉 PyTorch 框架,了解 NCCL、动/静态图、CUDA 算子实现等基本原理;
2、有算法基础了解,能部署并执行算法训练任务;
3、熟悉 Python、C/C++ 语言;
4、做过公司级训练平台搭建者优先;
5、接触过 Megatron、DeepSpeed 等大模型框架者优先;
6、快速学习能力,良好的团队协作精神和敬业精神。
20,861+ 岗位更新等你来订阅
一键订阅最新的岗位,每周送达
🎉恭喜你,订阅成功
继续订阅您可以在邮箱中随时取消订阅