Pymc 4.0 and variational inference

I got it to work. However, now I have to learn to use the output of the callbacks. Will report back later. :slight_smile: