dodgebc/jaxkd

Minimal JAX implementation of k-nearest neighbors using a k-d tree.

语言：Jupyter Notebook · GitHub 星标：60 · Fork：3

在 GitHub 上查看原仓库