Domain randomization on reset instead of initialization

[hartikainen]

DomainRandomizationVmapWrapper provides a clean way to randomize the model parameters at the initialization time. The fixed one-time randomization becomes increasingly sparse with the increasing randomization dimensionality. Being able to randomize at reset time could help with this. Is there any way to randomize the model on every reset?