mlx-graphs
M
Mlx Graphs
簡介 :
mlx-graphs是一個專為蘋果芯片設計的圖神經網絡(GNN)庫。它利用蘋果硬件的優勢,如統一內存架構,實現了在Mac設備上的高效GNN訓練和推理。主要優點有:1)利用GPU並行計算,在大數據集上實現高速GNN運算;2)利用統一內存,支持在GPU上直接處理大規模圖數據;3)無需設備間數據傳輸,簡化開發流程。用戶可以便捷地在Mac設備上處理大規模圖問題,實現高性能GNN模型訓練。
需求人群 :
["圖分類","節點分類","鏈接預測","圖生成"]
總訪問量: 474.6M
佔比最多地區: US(19.34%)
本站瀏覽量 : 51.1K
使用場景
使用內置的GCN進行化學分子圖的分類
實現基於GraphSAGE的節點分類模型
構建GAT層從Scratch進行鏈接預測任務
產品特色
高效的GNN訓練和推理
支持大規模圖數據處理
簡單的GNN層和模型構建
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase