Varun Agrawal

@btaba you mentioned you've been using SAC without any issues. Anything seem obvious about this off the top of your head? Additionally, how much GPU RAM do you use? I...

Interesting. I'll spend some time making a Colab example. If I can reproduce it there, that will be telling.

This Deepmind [example](https://colab.research.google.com/github/google-deepmind/mujoco/blob/main/mjx/tutorial.ipynb#scrollTo=xLiddQYPApBw), specifically the Quadruped section. PPO works great for me, but SAC throws the above error so this is a really curious issue.

@btaba here is a repo which I am able to reproduce the issue in: https://github.com/varunagrawal/locomotion Please let me know if you are able to get the same error. Again, PPO...

@btaba just updated it! Thanks for catching that and looking into this.

I have been facing the same issue. The good news is that the errors don't prevent pprof from working (at least for my case where I want to generate PDF...

This is being tackled in #17396

> Thanks for the PR. > > Can you comment on the practical usage and range of initial_alpha that has shown some improvement in your applications? Hi @btaba. I apologize...

@btaba I updated the PR comment to describe what I am trying to do better and I merged in the latest HEAD. Can you please review this when you get...