-
Hi,
We are currently trying to setup a simulation of a quadruped where each leg is designed like a five-bar-linkage system like this:
![image](https://github.com/google/brax/assets/32849595/1240d3…
-
Does Brax support computing gradient w.r.t. the physical parameters, like friction, stiffness, damping, et. al. Thanks!
-
I am following your rep to fine tune GPT-J on TPU. When I run "python3 device_train.py --config=YOUR_CONFIG.json --tune-model-path=gs://YOUR-BUCKET/step_383500/" with my bucket name and the config fil…
-
## Describe the bug
I quickly adapted the [benchmark for batched environments](https://github.com/pytorch/rl/blob/main/benchmarks/benchmark_batched_envs.py) to compare against native gymnasium envi…
-
Hi folks,
I am trying to use existing brax (0.9.3) wrappers with the new MjxEnv (Mujoco 3). I've set up my code following the examples from the mujoco jupyter notebook for defining an MjxEnv: https…
-
Hi,
so I understand that the physics simulation in the mujoco 3.0.0.0 is done on the GPU and is more faster, does that use the mujoco-mjx for this simulation.
From what I understand the MJX use…
-
I am looking to add sensory/depth perception to an agent so he is able to 'see' and navigate the environment before him. However, Brax does not seem to have a built-in method for computing depth such …
-
One of the most attractive things to me about brax is the possibility of highly parallelized sim-to-real applications for robotics. However, one of the issues that I'm facing right now that mujoco-men…
-
I meet a random leaked trace error when using humanoid env in my project. The error is thrown randomly. Therefore, I decide to use jax.config.update("jax_check_tracer_leaks", True) to find where the l…
-
I'd like to initialize an array according to a sharding, rather than initializing it on the default device and then moving it to the sharding. This is required when trying to instantiate arrays that a…