How to visualize when training like isaacGYM

[kassasin]

I want to visualize training process (not only rewards but also robot moving), does brax suport this>

[btaba]

Hi @kassasin , you can use the policy_params_fn callback to render videos for every eval that gets run