DifferentiableUniverseInitiative / flowpm

Particle Mesh simulation in TensorFlow
https://flowpm.readthedocs.io/
MIT License
90 stars 20 forks source link

Adds modification to cache growth ODE #57

Closed EiffL closed 3 years ago

EiffL commented 3 years ago

This PR adds a caching mechanism that ensures that the ODE is called only once, and then subsequent calls are interpolated from an interpolation table.

@dlanzieri please review this PR and let me know what you think :-)

A few observations on the modifications I've made:

I hope now everything should work fine