These packages are available as a package collection, usable in Xcode or SwiftPM.
GPU k-NN search and k-NN graph construction on Apple silicon, built on MLX
1 package.