jFoF: GPU Cluster Finding with Gradient Propagation

Benjamin Horowitz, Adrian E. Bayer

We present jFoF, a fully GPU-native Friends-of-Friends (FoF) halo finder designed for both high-performance simulation analysis and differentiable modeling. Implemented in JAX, jFo…