/mlx-GCN

MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).

Primary LanguagePythonMIT LicenseMIT

Stargazers