v0.1.2
Pre-release
Pre-release
Minor additions and fixes:
- Added gradient clipping and reward scaling to all algorithms
- Consistently use
jax.treesubmodule for support of newer jax versions - Experimental
rejax.compatsupport for gymnasium classic control environments - Fixed
NaNwhen calculating the log probability of an action that is on the extreme end of the action space withSquashedGaussianActor