sfd158

It seems the zero pose of this dataset is strange..

Is there any update on the release date for the code?

To reduce memory usage, there are some inplace operation in mjx.step. Inplace operation on intermediate matrics, such as X[0] += Y[0], will break back-propagation path. So jax.grad(mjx.step) doesn't work in...