Skip to main content

JZ-Tree: GPU friendly neighbour search and friends-of-friends with dual tree walks in JAX plus CUDA

Jens Stücker, Oliver Hahn, Lukas Winkler, Adrian Gutierrez Adame, Thomas Flöss · 2026

arXiv

Citation

@article{Stucker:2026wwx,
  author = {St{\"u}cker, Jens and Hahn, Oliver and Winkler, Lukas and Adame, Adrian Gutierrez and Fl{\"o}ss, Thomas},
  title = {{JZ-Tree: GPU friendly neighbour search and friends-of-friends with dual tree walks in JAX plus CUDA}},
  eprint = {2604.05885},
  archiveprefix = {arXiv},
  primaryclass = {cs.DC},
  month = {4},
  year = {2026},
  arxiv = {2604.05885}
}