-
Hey all,
I have been trying to load a few different URDFs in Brax on my own time. Brax seems to load them fine, but then the simulation seems to explode when I actually run it. No errors, but the r…
-
Hello,
I am fairly new to JAX and Equinox but I am running into a issue that I don't quite understand. I have the following loss function:
`@partial(jax.jit, static_argnames=('timeSteps'))
def …
-
Hi, I am trying to use the DMControlEnvs walker/quadruped with the default context value. But the gravity is zero even when the context has the correct value. Here are the packages I have installed in…
-
## 🐛 Bug Report
As reported in https://github.com/inseq-team/inseq/issues/232#issuecomment-1807001439, there might be an issue related to `jaxtyping` (#167).
```
AttributeError …
-
Hi,
I was excited to try out the new rendering added recently but it doesn't seem to work in a few cases.
I'm using MacOS 13.4, Python 3.9.6, Brax 0.9.1 and Jax 0.4.10.
Running
```
import…
-
I would like to add an object in the environment that is an obstacle with openings on the sides the agent can enter from. I thought of using a cylinder but I see it is not supported by brax. Do you ha…
-
-
1. `hugefoot` task missing in `scenarios/brax/ant.py`.
2. `alphas_q_update` needs be set near line 97 of `agents/subspace_agents.py`.
3. Learning rate for `optimizer_policy` and `optimizer_q` in `co…
-
### Describe the bug
wrapper conversion jax_to_numpy fails for NamedTuples in
As there are no specific handling of NamedTuples installed, the Iterable will be used
```
@jax_to_numpy.registe…
-
Let's say we have 2 environments:
- one `env` for collection with action spec with shape `[500,3,2]`, with [500] batch_size
- one `test_env` for testing with action spec with shape `[1,3,2]`, wi…