Skip to content

v0.1.2

Pre-release
Pre-release

Choose a tag to compare

@keraJLi keraJLi released this 23 May 11:48
· 41 commits to main since this release

Minor additions and fixes:

  • Added gradient clipping and reward scaling to all algorithms
  • Consistently use jax.tree submodule for support of newer jax versions
  • Experimental rejax.compat support for gymnasium classic control environments
  • Fixed NaN when calculating the log probability of an action that is on the extreme end of the action space with SquashedGaussianActor