Pymc 4.0 and variational inference

[cluhmann]

I suspect that you are looking for something like this:

pm.fit(n=1000, obj_optimizer=adam())

The various update methods can be found here.