-
Offloading BRAX to graphics card does not work on cip pool as expected. Look into this.
-
### Solution to issue cannot be found in the documentation.
- [X] I checked the documentation.
### Issue
The following works if you install `grpcio
-
### Proposal
Gymnasium already contains a large collection of wrappers, but we believe that the wrappers can be improved to
1. Support arbitrarily complex observation / action spaces. As RL has ad…
-
Hi,
In the `README.md` you show a visualization of an A1 robot. I had a look in the code but did not find any implementation for that environment.
I was simply wondering if you had any implement…
-
## Note
- [x] **I understand that irregular questions may be rejected.**
- [x] I understand that the issue raised here is **not** related to the front-end.
- [x] I have looked it up in opening i…
-
I am new to brax. When I try to perform an environment step using a jitted version of env.step (using code taken from the brax environments notebook), I get a TracerArrayConversionError. Any idea how …
-
Hi, thanks for all your work with Brax! I'm just getting into it, and am running into surprising behavior on a simple example:
```python
import jax
from brax import envs
env = envs.create('inv…
-
### Context
I am trying to use the inverted_pendulum environment in Brax 0.0.12 with google colab.
Even after 500 timesteps, the environnement never returns done (unless time limit is reached).
…
-
At some point when executing self.sys.step (line 224 of ant), brax tries to convert a traced array into a original numpy array, which JAX does not allow. I'm having a difficult time finding where exac…
-
https://github.com/google/brax/blob/ff3ff641097699703087e1dc0a7b6e8305d78270/brax/training/agents/ppo/networks.py#L46
I noticed something weird when comparing the mean of many sampled actions vs th…