What's Changed
* Fix by dirmeier in https://github.com/dirmeier/sbijax/pull/15
* Fix by dirmeier in https://github.com/dirmeier/sbijax/pull/16
* Improve training by dirmeier in https://github.com/dirmeier/sbijax/pull/17
* Add MH variants and MALA by dirmeier in https://github.com/dirmeier/sbijax/pull/18
* Add (slice) summary methods by dirmeier in https://github.com/dirmeier/sbijax/pull/19
* Cleanup by dirmeier in https://github.com/dirmeier/sbijax/pull/20
* Add SRNE by dirmeier in https://github.com/dirmeier/sbijax/pull/21
* Add documentation by dirmeier in https://github.com/dirmeier/sbijax/pull/22
* Move to TF iterators by dirmeier in https://github.com/dirmeier/sbijax/pull/23
* Add FMPE by dirmeier in https://github.com/dirmeier/sbijax/pull/25
* More documentation by dirmeier in https://github.com/dirmeier/sbijax/pull/26
* Add CMPE by dirmeier in https://github.com/dirmeier/sbijax/pull/27
**Full Changelog**: https://github.com/dirmeier/sbijax/compare/v0.1.0...v0.2.0