flowpm (pypi): Particle Mesh Simulation in TensorFlow
Keywords: cosmology|machine|learning