牛大妈在社招职位搜索PyTorch 有 2 条结果

招聘城市：深圳
岗位职责：
1.参与Pytorch等大模型深度学习框架的研发工作，重点负责GPU及异构芯片的适配与性能优化；
2.深入研究框架引擎内部机制，参与或主导前沿大模型相关技术的探索与实践；
3.与团队合作，持续提升系统的稳定性和扩展性，确保系统能够满足大规模深度学习任务的需求。
岗位要求：
1.精通PyTorch框架，在Pytorch框架上具备大模型的分布式训推经验者优先；
2.掌握并行计算、CUDA编程、网络通信、系统优化及集群硬件架构等HPC相关知识；
3.具备出色的编程能力，熟练使用Python，掌握C++、数据结构与算法设计，熟悉Linux/Unix系统及Shell编程，熟练使用Git进行版本控制；
4.拥有三年以上AI分布式系统研发经验，具备成熟的系统…

招聘城市：上海，北京
岗位职责：
1、参与大规模分布式训练框架（基于PyTorch等）的研发与优化，重点解决千卡乃至万卡级别集群的扩展性、容错与性能瓶颈问题；
2、负责训练系统中关键组件的设计与开发，包括但不限于大规模分布式参数服务器、高性能嵌入向量（Embedding）训练与推理、高效的优化器实现等；
3、深入系统底层，利用高性能网络（如RoCE v2, InfiniBand）、RDMA技术以及NVMe SSD存储，优化数据读取、通信和Checkpoint存储等关键路径，实现极致的端到端训练性能；
4、与算法团队紧密合作，理解前沿模型（如大语言模型、推荐系统、多模态模型）对训练基础设施的需求，并将其转化为系统级的创新与优化；
5、负责GPU/XPU计算资源的精细调度与…