JAX Version Strangeness
luchris429 opened this issue · comments
There are strange gaps in speed and performance of JAX v0.4.8 and JAX v 0.4.13. Currently investigating. I would recommend sticking with <=0.4.8 for now.
luchris429 / purejaxrl
Really Fast End-to-End Jax RL Implementations