-
https://github.com/google/brax/blob/280a1c50fa021b6c17a2a3347fea43a2887382bc/brax/v2/math.py#L278
For people who want to calculate gradients over environment steps, this while loop is a bit annoyin…
-
Hi, I have an extremely strange bug that is slightly hard to reproduce :)
I am trying to integrate brax with SAC and want to use an ensemble version of the algorithm. When I use the ensemble, the bra…
-
Probably harmless but just in case:
```
Traceback (most recent call last):
File "ppo_test.py", line 45, in testTrain
self.assertGreater(metrics['eval/episode_reward'], 100 * 0.995)
Assert…
-
**Version** : 0.2.7.8
**Date** : 28 June 2018
-
Hi all,
Is rendering without notebooks supported? I personally don't like using notebooks and am much more used to working with e.g. GLUT windows. It seems that wrapping brax environment into `gym`…
-
### Intro
Hi!
I am a graduate, I use MuJoCo for my research on RL.
### My setup
mujoco version: 3.2.5
python api
64 bit
Ubuntu 24.04.1 LTS
RTX 2060 super, 8GB @ 2010 MHz
### What'…
-
Lots of other frameworks seem to be switching to gymnasium. It'd be nice to also have a gymnasium wrapper in brax.
-
The reference `Ant` and the 2 `Humanoid`s implementations do not include them, but this issue:
https://github.com/google/brax/issues/254
Indicates that they might be implemented which is it?
If…
-
For domain randomization it is not particularly easy to `vmap` over different `System` values. For example the `gravity` values, or the `elasticity`. Preferably you should be able to do this in `env.r…
-
If you go to:
https://colab.research.google.com/github/google/brax/blob/main/notebooks/composer/composer.ipynb
and choose `humanoid_run` as the environment, the next cell throws the following error:…