siwei0729

It improves the training speed about 2x to 4x faster, tested on a 2080Ti machine. Since the online training dataset is small, we can store all the data in the...

Brax Halfcheetah Exploding Gradients

1

Hi, I'm working on a project that uses differentiable dynamics. However, for the task halfcheetah, I'm having problems with the gradient explosion. I have created a [repo](https://github.com/siwei0729/brax_halfcheetah_gradient) to reproduce this...