Uniform vectorization #117
Description
Right now vector envs are implemented for brax internally (bc they have to be for brax 0.9), but would need to be external fo the other envs. This isn't great since it means we need to deal with vec envs in a case by case manner and should be corrected. Brax updating could already help.