Adding momentum to optimizers for variational inference


I’m trying out different optimizers for VI using the obj_optimizer keyword like this:

 advi_res =, obj_optimizer=pm.adam(), callbacks=[tracker])

I see here that it’s possible to add momentum to each optimizer but I can’t figure out the API for doing that. I’d appreciate some help.