btaba

### Discussed in https://github.com/google/brax/discussions/403 Originally posted by **eleninisioti** October 11, 2023 There is a notebook that explains how to save and load models (https://github.com/google/brax/blob/main/notebooks/training.ipynb) but there testing happens right after...

> Are you thinking to use boxes for now? Go for it! If you have a lot of boxes, consider adding these fields to your model > > https://github.com/google-deepmind/mujoco/blob/a8db22f0d077aee86f771808fd24ea60a148c93f/mjx/mujoco/mjx/test_data/shadow_hand/right_hand.xml#L8-L11 >...

See https://github.com/google/brax/discussions/536