社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  机器学习算法

JittorGeometric 2.0:基于国产深度学习框架 J-20250907135911

爱可可-爱生活 • 11 月前 • 868 次点击  

2025-09-07 13:59

JittorGeometric 2.0:基于国产深度学习框架 Jittor 的图机器学习库,专注于高效、灵活和可扩展的图神经网络研究与应用。

• 🚀 JIT 编译加持,实现动态代码修改,无需预编译,显著提升开发效率。
• ⚡ CuSparse 加速的稀疏矩阵运算,优化大规模图计算性能。
• 🎯 超 40 种模型涵盖经典、谱域、动态图、分子图及图Transformer,覆盖最新研究前沿。
• 📊 内置Planetoid、OGB、Reddit等主流图数据集加载器,方便快速实验。
• 🔄 支持多GPU多节点分布式训练,满足企业级规模扩展需求。
• 🌊 动态图事件驱动并行处理,适应时变网络建模。
• 📦 支持大规模图的高效mini-batch训练,降低内存压力。
• 🎛️ 新增Ascend NPU支持,扩展硬件适配面,兼容多样计算平台。
• 简洁Python API,配套详尽示例与文档,助力快速上手与深度定制。

这套国产图学习工具包,突破传统GNN性能瓶颈,助力科研与工业场景双重落地,极具长期价值参考意义。

了解详情🔗github.com/AlgRUC/JittorGeometric
#图神经网络# #深度学习# #Jittor# #机器学习# #分布式训练#
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/186472