AutoDelta is used to do MAP inference in Pyro. It would be nice to have it in NumPyro too.

References

• Pyro AutoDelta implementation
• AutoNormal implementatio

Hi @JavierAntoran @stratisMarkou,

First of all, thanks for making all of this code available - it's been great to look through!

Im currently spending some time trying to work through the Weight Uncertainty in Neural Networks in order to implement Bayes-by-Backprop. I was struggling to understand the difference between your implementation of `Bayes-by-Bac