JZ-FMM: GPU-native differentiable N-body simulations with the Fast Multipole Method
N-body simulations are an essential tool for modelling the evolution of gravitating systems. Differentiable N-body simulations allow solving complicated reconstruction problems by connecting observations of evolved systems to simple informative priors on their initial conditions. While so far most reconstruction effort...