Metadata-Version: 2.4
Name: jztree
Version: 1.0.3
Summary: A framework for Morton z-order trees for JAX with CUDA backend
Author-email: Jens Stuecker <jens.stuecker@univie.ac.at>
License-Expression: MIT
Project-URL: Homepage, https://jstuecker.github.io/jztree/
Project-URL: Issues, https://github.com/jstuecker/jztree/issues
Requires-Python: >=3.11
Description-Content-Type: text/markdown
Requires-Dist: jax>=0.8.0
Requires-Dist: numpy>=2.2.6
Provides-Extra: cuda12
Requires-Dist: jztree-cu12==1.0.1; extra == "cuda12"
Provides-Extra: cuda13
Requires-Dist: jztree-cu13==1.0.1; extra == "cuda13"

# jz-tree
jz-tree offers a framwork for GPU-friendly implementations of tree algorithms in jax (with a CUDA backend). Currently nearest neighbour search and friends-of-friends are supported and deliver top performance.

Install either with
```bash
pip install jztree[cuda12]
```
or 
```bash
pip install jztree[cuda13]
```
(depending on your jax/cuda setup).

For detailled installation instructions, please check [the documentation](https://jstuecker.github.io/jztree/installation.html) or the [code repository](https://github.com/jstuecker/jztree).
