🐛 Bug

Using DDP after #18953, with a model where not all model parameters are used in a particular iteration, and a dict return value (from forward), doesn't work.

This is a known problem and we can easily solve by adding a smarter return value scanner that deals with dicts as well (and recurses into them?).