JZ-Tree: GPU friendly neighbour search and friends-of-friends with dual tree walks in JAX plus CUDA
El artículo presenta JZ-Tree, una implementación de código abierto en JAX y CUDA que utiliza una jerarquía de árbol basada en planos de orden Morton para superar los problemas de divergencia de hilos y acceso a memoria en GPUs, logrando mejoras de rendimiento de más de un orden de magnitud en búsquedas de vecinos más cercanos y agrupamiento "friends-of-friends" para grandes conjuntos de datos.