File train_brax.py

mbd/rl/train_brax.py:None–None

import os
import functools
from datetime import datetime
from brax import envs