Public code repository for the paper "Model-free learning of probability flows: Elucidating the nonequilibrium dynamics of flocking" (https://arxiv.org/abs/2411.14317).
Contains an implementation of the associated learning algorithm, dataset generation code, and an API for common active matter systems such as active OU and Vicsek particles.
Model checkpoints available upon request.