Pyro backend VB

[pdh21]

Expanding to more complicated models is going to require moving to Variational inference. Pyro looks like the best backend to use to be able to implement this

[pdh21]

Standard SPIRE model test

Time comparison

No subsampling

fit_pyro=SPIRE.all_bands([prior250,prior350,prior500],n_steps=500,lr=0.001)

CPU times: user 2min 9s, sys: 20.6 s, total: 2min 29s
Wall time: 1min 6s

with subsampling

fit_pyro=SPIRE.all_bands([prior250,prior350,prior500],n_steps=500,lr=0.001)

CPU times: user 1min 2s, sys: 7.24 s, total: 1min 9s
Wall time: 37.7 s