I’m not aware of any but wouldn’t the Optimizer base class work there?
As far as I understand, the trouble with modules comes mainly from the fact that it is not a proper subclassing because the signatures of forward are incompatible. I would naively expect that this should be better with optimizers.
Best regards
Thomas