brax

brax copied to clipboard

Hello Brax community, This post presents a prototype for a real-time, web-based visualizer for Brax training, developed as part of [Google Summer of Code 2025](https://summerofcode.withgoogle.com/programs/2025/projects/Xm0toJHl). This work aims to address...

bsyh

Calculating gradient for system identification gives jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: batch in most minor dimension

1

## Description I am trying to find the optimal system parameters (damping/mass etc.) to minimize the MSE between real-sim trajectories. When I tried to get the gradient from the loss...

jongyaoY

Hi Brax team, JummerCloth again. Thanks for adding the parametric action distribution! I’d like to suggest another small but useful feature enhancement to the `ppo.train` function: adding a new argument...

JummerCloth