search

furcelay / gigalens

Gradient Informed, GPU Accelerated Lens modelling (GIGALens) -- a package for fast Bayesian inference on strong gravitational lenses.
https://giga-lens.github.io/gigalens
MIT License
0 stars 0 forks source link

Sparse convolution #4

Open furcelay opened 5 months ago

furcelay commented 5 months ago

Use sparse convolution with the PSF in the simulation to avoid conversion to a dense matrix. This can save memory and be much faster.

furcelay commented 5 months ago

Seems that there is no implementation of submanifold sparse convolution on TensorFlow or JAX (only Pytorch).