Create 0.0.1b0 release.

Changelog:

Examples:

- Added VAE example.
- Added pruning example (https://arxiv.org/abs/1710.01878).
- MNIST example uses 300-100-10 MLP.
- Updated imagenet dataset to return correctly scaled examples.

Breaking changes:

- State arg to `hk.transform` dropped in favor of `transform_with_state`.
- Decay argument is now required in `BatchNorm`.

Features:

- Added `hk.maybe_next_rng_key()`.
- BatchNorm and LayerNorm speed improvements.
- Added support for partition/filter/merge params.
- Haiku now allows running with `jax_numpy_rank_promotion`.

Experimental features:

- `hk.experimental.to_dot` - experimental visualisation support.
- `hk.experimental.lift` - experimental purification support.

Usability:

- Improved error message when RNG arg is not and RNG.
- Improved documentation.
- Improved test coverage.

PiperOrigin-RevId: 303113097
Change-Id: Ia08a3a02be9d48a5e4261bb5072e2b9f78b66ae0

README.md

@@ -21,11 +21,10 @@ library for [TensorFlow](https://github.com/tensorflow/tensorflow).
 **Disambiguation:** if you are looking for Haiku the operating system then
 please see https://haiku-os.org/.
 
-NOTE: Haiku is currently **alpha**. A number of researchers have tested Haiku
+NOTE: Haiku is currently **beta**. A number of researchers have tested Haiku
 for several months and have reproduced a number of experiments at scale. Please
-feel free to use Haiku, but be sure to test any assumptions and to
-[let us know](https://github.com/deepmind/dm-haiku/issues) if things don't look
-right!
+feel free to use Haiku, and
+[let us know](https://github.com/deepmind/dm-haiku/issues) if you have issues!
 
 ## Overview
 
@@ -403,7 +402,7 @@ To cite this repository:
   author = {Tom Hennigan and Trevor Cai and Tamara Norman and Igor Babuschkin},
   title = {{H}aiku: {S}onnet for {JAX}},
   url = {http://github.com/deepmind/dm-haiku},
-  version = {0.0.1a0},
+  version = {0.0.1b0},
   year = {2020},
 }
 ```

haiku/__init__.py

@@ -87,7 +87,7 @@
 from haiku._src.typing import Params
 from haiku._src.typing import State
 
-__version__ = "0.0.1a0"
+__version__ = "0.0.1b0"
 
 __all__ = (
     "AvgPool",

setup.py

@@ -61,7 +61,7 @@ def _parse_requirements(requirements_txt_path):
     zip_safe=False,
     # PyPI package information.
     classifiers=[
-        'Development Status :: 3 - Alpha',
+        'Development Status :: 4 - Beta',
         'Intended Audience :: Developers',
         'Intended Audience :: Education',
         'Intended Audience :: Science/Research',