DifferentiableUniverseInitiative / flowpm

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

Implement distributed lensing #81

Open EiffL opened 3 years ago

EiffL commented 3 years ago

This issue is to track the development of a distributed version of our FlowPM lensing implementation.

We essentially need to create a distributed version of these two funnctions:

The first one should be easily adaptable from the distributed versions of cic_paint which can be found here: https://github.com/DifferentiableUniverseInitiative/flowpm/blob/9145d0c846d886f2cb34955f2267aa8cc9da31da/flowpm/mesh_utils.py#L125

The second one is a tiny bit more sophisticated because it will involve a distributed interpolation operation and this might require some thinking