v0.1.2

Remove jax[cuda] dependency in default non-cuda install.

v0.1.1

Add optional jaxkd-cuda dependency and integrate as a cuda=True argument to core functions.

v0.1.0

Changes:

• count_neighbors and query_neighbors now require r and k as keyword arguments for clarity
• New k-means implementation in jaxkd.extras
• Fix bugs encountered in x64 mode

Thanks to @NeilGirdhar for contributions to the k-means implementation and overall improvements to code style!

v0.0.3

Add brute force pairwise distance versions to extras.py for convenient comparison.

v0.0.2

New count_neighbors operation and simple k_means implementation.

v0.0.1

Initial release.