You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
vmap is not compatible with torchinterp1d or keops. Both use torch.autograd.Function, which is not compatible with vmap by default. This will be a problem if you want to e.g. vmap over a bunch of NFWs for simulating or computing magnifications, since they may use torchinterp1d via FlatLambdaCDM. This is fixable, but requires modifying torchinterp1d and keops. I've opened issues in both projects' repos.
This isn't really our responsibility to fix, but I wanted to document this since it's good to be aware of this restriction.
The text was updated successfully, but these errors were encountered:
We no longer depend on torchinterp1d and only have limited compatibility with keops. It would be good to open another issue specifically for keops compatibility.
vmap
is not compatible withtorchinterp1d
orkeops
. Both usetorch.autograd.Function
, which is not compatible withvmap
by default. This will be a problem if you want to e.g.vmap
over a bunch ofNFW
s for simulating or computing magnifications, since they may usetorchinterp1d
viaFlatLambdaCDM
. This is fixable, but requires modifyingtorchinterp1d
andkeops
. I've opened issues in both projects' repos.This isn't really our responsibility to fix, but I wanted to document this since it's good to be aware of this restriction.
The text was updated successfully, but these errors were encountered: