Pytorch框架研发高级工程师（深圳/北京/上海/杭州） 深圳 分享

分享岗位

方式1:复制岗位链接

方式2:分享岗位海报

手机扫描二维码分享

收藏 TEG 点击了解更多BG信息 技术 一年以上工作经验 更新于年06月12日

在腾讯，后台开发工程师不仅是“又快又稳”的问题解决专家，更是生态共创者。你将与技术团队一同沉淀优质代码，让它成为我们共有的宝贵资产。在不同的业务场景和技术发展阶段，你的架构思维也将帮助更多协作团队拓展新的思考。我们也珍视你的挑战精神，同时欢迎你一起参与团队愿景、文化和产品方向的探讨。

岗位职责

1.参与开发优化pytorch基础框架，支持公司内业务通用训推工程需求；
2.联合业务进行模型训推效率分析优化、精度对齐、效果验证等工作；
3.参与基础框架的国产卡适配接入，提升框架异构算力使用效率。

岗位要求

1.熟练掌握pytorch框架，可对训练和推理任务进行性能分析和优化，具备框架特性开发的能力；
pile（TorchDynamo、AOTAutograd、PrimTorch 和 TorchInductor）、torch.export、torch DDP和FSDP的流程和实现机制；
3.有搜广推、生成式推荐、LLM、ViT、AST、DiT、Bert模型业务精调和工程优化经验者优先；
4.有pytorch开源社区代码贡献者优先；
5.良好的基础编程能力、沟通能力、解决问题能力。