Jumanji v0.3.1

Release Notes

In this release, we update the dependencies to support the latest jax (and chex) version and optimize the environment speed of Game2048.

What's Changed

• fix(examples): port notebook to colab by @clement-bonnet in #169
• ci: update to latest jax and chex by @clement-bonnet in #174
• build: remove jaxlib by @clement-bonnet in #175
• docs: update readme citation by @clement-bonnet in #176
• feat(2048): environment performance improvements by @aar65537 in #172
• build: bump version to 0.3.1 by @clement-bonnet in #177

Full Changelog: v0.3.0...v0.3.1