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.
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.