[Feature]: Env-wise gravity vector

[nootnootpengupengu]

trafficstars

What feature or enhancement are you proposing?

The ability to set a unique gravity vector for each of my simulation environments, not a global vector for all envs.

Motivation

Part of my domain randomization scheme requires varying the gravity vector between all envs during intraining. It is currently not possible to do so as it's a global parameter in SimOptions.

Potential Benefit

Improved policy training and robustness due to increased resilience from different gravity force.

What is the expected outcome of the implementation work?

The ability to set a SimOptions.gravity = gravity_tensor, where gravity_tensor.shape = (num_envs, 3)

Additional information

No response