mlx-graphs
M
Mlx Graphs
简介 :
mlx-graphs是一个专为苹果芯片设计的图神经网络(GNN)库。它利用苹果硬件的优势,如统一内存架构,实现了在Mac设备上的高效GNN训练和推理。主要优点有:1)利用GPU并行计算,在大数据集上实现高速GNN运算;2)利用统一内存,支持在GPU上直接处理大规模图数据;3)无需设备间数据传输,简化开发流程。用户可以便捷地在Mac设备上处理大规模图问题,实现高性能GNN模型训练。
需求人群 :
["图分类","节点分类","链接预测","图生成"]
总访问量: 474.6M
占比最多地区: US(19.34%)
本站浏览量 : 53.0K
使用场景
使用内置的GCN进行化学分子图的分类
实现基于GraphSAGE的节点分类模型
构建GAT层从Scratch进行链接预测任务
产品特色
高效的GNN训练和推理
支持大规模图数据处理
简单的GNN层和模型构建
AIbase
智启未来,您的人工智能解决方案智库
© 2025AIbase