Add multiple snapshots observable

eelregit / pmwd

Differentiable Cosmological Forward Model

BSD 3-Clause "New" or "Revised" License

69 stars 16 forks source link

Open Yucheng-Zhang opened 1 year ago

Yucheng-Zhang commented 1 year ago

We add support for recording multiple snapshots at given times using cubic Hermite interpolation.
The user specifies the times through conf.a_snapshots.
For the forward evolution, we carry the previous ptcl, such that the interpolation can be done in one step.
For the backward adjoint evolution, the interpolation has to be split into two steps.