Dopamax 0.2.1

Changes

• Add SAC, DDPG, and TD3 algorithms.
• Add support for prioritized replay buffers.
• Add support for auto-loading any gymnax environment.
• Various bug fixes.
• Switch to using pyproject.toml.
• Adopt flashbax for replay buffers.

Dopamax 0.2.0

Changes

• Add support of for AlphaZero.
• Add PGX environments to be used with AlphaZero.
• Remove use of JAX callbacks for logging (previous implementation wasn't compatible with TPUs). Training loop now exits out of XLA execution to execute callbacks.
• Lots of refactoring/restructuring of codebase.
• Increase JAX dependency version.

Dopamax 0.1.0

Initial release.