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
pip install jztree[cuda12]
or
pip install jztree[cuda13]
(depending on your jax/cuda setup).
For detailled installation instructions, please check the documentation or the code repository.
1 maintainer
Metadata
1 maintainer
Release files for jztree 1.1.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jztree-1.1.0-py3-none-any.whl | Python 3 | none | any | Details |
Release files / jztree-1.1.0-py3-none-any.whl
| Download URL | jztree-1.1.0-py3-none-any.whl |
|---|---|
| Size | 173.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3824902f6a5a718b221459aa0943f3f7edfef8f5a8403edf091d2c987f7748d0
|
|
BLAKE2b-256 checksum How to use checksums |
35697792fc36ad22033897910d3461a4df6f276a92fca5e08f542cb5f18c3645
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.12.11
|
