brax
brax copied to clipboard
google
Massively parallel rigidbody physics simulation on accelerator hardware.
Hi~ I am currently following a physics-informed work repo [Physics-informed nn](https://github.com/wuwushrek/physics_constrained_nn/issues/1) which use brax as a benchmark. In there they use `active_pos` and 'active_rot' to remove the inactive state dimension...
I've been digging into Brax as a potential alternative to some modified dm_control enviornments I've been using and am really loving the speedup! That said, I feel like I've run...
### 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)....
Due to a "can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first." error in pyplot, but fixing this raises another error.
Hi, Trying to create a grasping simulation. As per our notebook: [https://github.com/nomagiclab/brax-grasping-sim/blob/master/notebooks/lifting_check.ipynb](https://github.com/nomagiclab/brax-grasping-sim/blob/master/notebooks/lifting_check.ipynb), generating 400 steps takes ~260 seconds on our local machine, ~500 seconds on google colab. What could cause...
Had a conversation with @jkterry1 on https://github.com/openai/gym/issues/2366, and it appears brax would also be a great alternative for the mujoco envs replacement. To help with this transition. I made an...
Dear Brax team, Since Brax is fully differentiable, I thought it'd be possible to use it like DiffTaichi or GradSim for system identification (e.g. determining the mass of an object...
Brax's rendering is CPU-only at the moment. For agents with vision, it would be useful for rendering to happen on accelerator so that training can stay fast. [Vispy](https://vispy.org/) may be...
Hi, just found this project. Don't really have a concrete issue, but some design questions id like to discuss. Ive recently started work on a similar project in JAX, that...
