First public release

What's Changed

• Bump versions for publishing by @StannisZhou in #63
• Example notebook with PMAP sampling of RBMs trained on MNIST digits by @StannisZhou in #80
• Use functools.partial instead of jax.partial by @StannisZhou in #83
• First pass for speeding up graph and evidence operations by @StannisZhou in #84
• Moving to a functional interface by @StannisZhou in #88
• Update README in preparation for making repo public by @StannisZhou in #89
• Pre commit ci test by @NishanthJKumar in #90
• [pre-commit.ci] pre-commit autoupdate by @pre-commit-ci in #91
• Update dependency requirements to be less aggresive by @StannisZhou in #92
• adds codecov badge to README! by @NishanthJKumar in #94
• Fix GPU memory leak that came up in RCN example by @StannisZhou in #97
• [pre-commit.ci] pre-commit autoupdate by @pre-commit-ci in #95
• Docs update by @NishanthJKumar in #98
• fixes bug where wrong conf.py path was specified by @NishanthJKumar in #99
• Rtd warning fix by @NishanthJKumar in #100
• includes minor changes to README and documentation by @NishanthJKumar in #101
• Bump up version; Fixes for docs by @StannisZhou in #102

Full Changelog: v0.2.0...v0.2.1