-
Hi!
I just tried to use this notebook: https://github.com/Denys88/rl_games/blob/master/notebooks/brax_training.ipynb and an import error occurred on this:
```python
from brax import jumpy as jp…
-
Hi, thanks for your work!
I was just going through the paper, and noticed a possible shape mismatch in the formulas, and wanted to clarify.
In Equations 1 & 2, the product shown is `BAx` and `BRAx…
-
Thought id mention; there is a brax implementation of SHAC [here](https://github.com/google/brax/pull/262#issuecomment-1387428472)
I suppose its hard to compare directly since the envs are not iden…
-
Hi! I am interested in training the ant robot using the CMA-ES strategy.
I tried running the notebook examples/07_brax_control.ipynb and the performances I get are off. When running it as it is, wi…
-
I'm currently doing a project related to transferring control policies from simulator to simulator. This lead me to try to standardize action and observation spaces between brax and mujoco.
I imple…
-
Todo: Look into adding support for Google's cool-looking [Brax framework](https://github.com/google/brax) to get massively parallelized [mujoco-like environments](https://github.com/google/brax/blob/m…
-
### 🚀 Feature
It would be nice to have a wrapper that ingested gymnasium.vector.VectorEnv and gave back a VecEnv.
### Motivation
I want to do highly parallelized hardware accelerated simulation. Th…
-
Hi,
I'm working on a project that uses differentiable dynamics. However, for the task halfcheetah, I'm having problems with the gradient explosion. I have created a [repo](https://github.com/siwei0…
-
I'm looking to simulate a tendon within Brax so that I can model a dexterous hand and train an RL agent to move it naturally.
This works in Mujoco right now by treating the muscle connected to the …
-
When I try to use a 4-gpus machine to run the Analytic policy gradients training in parallel, it reports an AssertionError in `brax/training/agents/apg/train.py` line 255. Seems that it is because `t…