https://hgpu.org/?p=23849
FlowPM: Distributed TensorFlow Implementation of the FastPM Cosmological N-body Solver