What's Changed
* [Docs] Add API reference by sotetsuk in https://github.com/sotetsuk/pgx/pull/770
* [CI] Add Python 3.9, 3.10 support to `setup.py` and run CIs by sotetsuk in https://github.com/sotetsuk/pgx/pull/785
* Revert "[Experimental] add `state._info` to `State`" by sotetsuk in https://github.com/sotetsuk/pgx/pull/788
* [TicTacToe] Change observation dims by sotetsuk in https://github.com/sotetsuk/pgx/pull/789
* [API] Use chex's dataclass instead of flax's dataclass by sotetsuk in https://github.com/sotetsuk/pgx/pull/787
**Full Changelog**: https://github.com/sotetsuk/pgx/compare/v0.3.2...v0.4.0