Adds base.py, which implements both a base MjxEnv class for environment management w/ Brax, and a State class to represent the RL state. This is copy-pasted from the mjxtutorial notebook.
Adds flux and brax as dependencies since they're intertwined with the current "idiomatic" way of doing RL in mjx, which uses the algorithms implemented in brax.
Does what it says on the tin. Major changes:
base.py, which implements both a baseMjxEnvclass for environment management w/ Brax, and aStateclass to represent the RL state. This is copy-pasted from themjxtutorial notebook.fluxandbraxas dependencies since they're intertwined with the current "idiomatic" way of doing RL inmjx, which uses the algorithms implemented inbrax.