jaxkd

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

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jaxkd

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.1.2 2025-07-15
0.1.1 2025-06-18
0.1.0 2025-05-26
0.0.3 2025-05-19
0.0.2 2025-05-18
0.0.1 2025-05-17

Issues with this package?

Page last updated 2025-07-17 22:52:43 UTC