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 (CoRR, 2026)