Merge zero_optimizers and zero_optimizer into a single function

We should merge zero_optimizers and zero_optimizer functions into a single one. See https://gitlab.com/teklia/atr/dan/-/blob/2adb6ba8e9f7698b3f7968292a87c6dde920f42b/dan/manager/training.py#L533