jstuecker/jztree

A framework for GPU-friendly implementations of tree algorithms -- such as kNN and FoF -- in jax (with a CUDA backend)

语言：Jupyter Notebook · GitHub 星标：55 · Fork：6

在 GitHub 上查看原仓库