Ilya Kostrikov

I'm not sure if ```python obs = obs.at[step].set(x) ``` is indeed in-place inside of jit. I think in this specific case it still creates a new array. I think it's...

@vwxyzjn yes, I think it's either for arrays created inside of jit or donated arguments.

I think that it happened because they changed default parameters for some functions recently. Try to check all "sum", "mean" and similar functions and set keep dim to True.

Sorry, I didn't have enough time recently to fix problems with this code. Could you submit a PR with this fix?

I have a similar problem. Sometimes it renders images correctly, but sometimes it renders only the background image (see the [video](https://drive.google.com/file/d/12JDiXLn3tHMrVqNizbu3M0kYb1mqbqNU/view?usp=sharing)). This issue is non-deterministic, and the video might be...

@saran-t `DISABLE_RENDER_THREAD_OFFLOADING=1` doesn't resolve the problem for me either.

NAF might be a little bit outdated (I'm not planning to maintain this repository). Is there a specific reason to use NAF? I personally find it easier to tune soft...

NAF is extremely sensitive to hyper parameters, so I didn't try to benchmark it.

What version of pytorch do you use?

Did you install it with anaconda or build from sources?