LICENSE
MANIFEST.in
README.md
setup.py
dm_sonnet.egg-info/PKG-INFO
dm_sonnet.egg-info/SOURCES.txt
dm_sonnet.egg-info/dependency_links.txt
dm_sonnet.egg-info/not-zip-safe
dm_sonnet.egg-info/requires.txt
dm_sonnet.egg-info/top_level.txt
docs/conf.py
docs/ext/link_tf_api.py
docs/ext/link_tf_api_test.py
examples/functional_mlp_mnist.py
examples/simple_mnist.py
examples/simple_mnist_test.py
sonnet/__init__.py
sonnet/distribute.py
sonnet/functional.py
sonnet/initializers.py
sonnet/mixed_precision.py
sonnet/optimizers.py
sonnet/pad.py
sonnet/regularizers.py
sonnet/nets/__init__.py
sonnet/nets/resnet.py
sonnet/src/__init__.py
sonnet/src/axis_norm.py
sonnet/src/axis_norm_test.py
sonnet/src/base.py
sonnet/src/base_test.py
sonnet/src/batch_apply.py
sonnet/src/batch_apply_test.py
sonnet/src/batch_norm.py
sonnet/src/batch_norm_test.py
sonnet/src/bias.py
sonnet/src/bias_test.py
sonnet/src/build.py
sonnet/src/build_test.py
sonnet/src/conv.py
sonnet/src/conv_test.py
sonnet/src/conv_transpose.py
sonnet/src/conv_transpose_test.py
sonnet/src/custom_getter.py
sonnet/src/custom_getter_test.py
sonnet/src/deferred.py
sonnet/src/deferred_test.py
sonnet/src/depthwise_conv.py
sonnet/src/depthwise_conv_test.py
sonnet/src/dropout.py
sonnet/src/dropout_test.py
sonnet/src/embed.py
sonnet/src/embed_test.py
sonnet/src/group_norm.py
sonnet/src/group_norm_test.py
sonnet/src/initializers.py
sonnet/src/initializers_test.py
sonnet/src/leaky_clip_by_value.py
sonnet/src/leaky_clip_by_value_test.py
sonnet/src/linear.py
sonnet/src/linear_test.py
sonnet/src/metrics.py
sonnet/src/metrics_test.py
sonnet/src/mixed_precision.py
sonnet/src/mixed_precision_test.py
sonnet/src/moving_averages.py
sonnet/src/moving_averages_test.py
sonnet/src/once.py
sonnet/src/once_test.py
sonnet/src/pad.py
sonnet/src/pad_test.py
sonnet/src/parallel_linear.py
sonnet/src/parallel_linear_test.py
sonnet/src/recurrent.py
sonnet/src/recurrent_test.py
sonnet/src/regularizers.py
sonnet/src/regularizers_test.py
sonnet/src/reshape.py
sonnet/src/reshape_test.py
sonnet/src/scale_gradient.py
sonnet/src/scale_gradient_test.py
sonnet/src/sequential.py
sonnet/src/sequential_test.py
sonnet/src/test_utils.py
sonnet/src/types.py
sonnet/src/utils.py
sonnet/src/utils_test.py
sonnet/src/conformance/__init__.py
sonnet/src/conformance/api_test.py
sonnet/src/conformance/build_test.py
sonnet/src/conformance/checkpoint_test.py
sonnet/src/conformance/copy_test.py
sonnet/src/conformance/descriptors.py
sonnet/src/conformance/descriptors_test.py
sonnet/src/conformance/distribute_test.py
sonnet/src/conformance/doctest_test.py
sonnet/src/conformance/function_test.py
sonnet/src/conformance/goldens.py
sonnet/src/conformance/goldens_test.py
sonnet/src/conformance/keras_test.py
sonnet/src/conformance/optimizer_test.py
sonnet/src/conformance/pickle_test.py
sonnet/src/conformance/saved_model_test.py
sonnet/src/conformance/tensorflow1_test.py
sonnet/src/conformance/xla_test.py
sonnet/src/conformance/checkpoints/generate.py
sonnet/src/distribute/__init__.py
sonnet/src/distribute/distributed_batch_norm.py
sonnet/src/distribute/distributed_batch_norm_test.py
sonnet/src/distribute/replicator.py
sonnet/src/distribute/replicator_test.py
sonnet/src/distribute/replicator_test_utils.py
sonnet/src/functional/__init__.py
sonnet/src/functional/haiku.py
sonnet/src/functional/haiku_test.py
sonnet/src/functional/jax.py
sonnet/src/functional/jax_test.py
sonnet/src/functional/optimizers.py
sonnet/src/functional/optimizers_test.py
sonnet/src/functional/utils.py
sonnet/src/nets/__init__.py
sonnet/src/nets/cifar10_convnet.py
sonnet/src/nets/cifar10_convnet_test.py
sonnet/src/nets/mlp.py
sonnet/src/nets/mlp_test.py
sonnet/src/nets/resnet.py
sonnet/src/nets/resnet_test.py
sonnet/src/nets/vqvae.py
sonnet/src/nets/vqvae_test.py
sonnet/src/nets/dnc/__init__.py
sonnet/src/nets/dnc/control.py
sonnet/src/nets/dnc/control_test.py
sonnet/src/nets/dnc/read.py
sonnet/src/nets/dnc/read_test.py
sonnet/src/nets/dnc/util.py
sonnet/src/nets/dnc/util_test.py
sonnet/src/nets/dnc/write.py
sonnet/src/nets/dnc/write_test.py
sonnet/src/optimizers/__init__.py
sonnet/src/optimizers/adam.py
sonnet/src/optimizers/adam_test.py
sonnet/src/optimizers/momentum.py
sonnet/src/optimizers/momentum_test.py
sonnet/src/optimizers/optimizer_tests.py
sonnet/src/optimizers/optimizer_utils.py
sonnet/src/optimizers/rmsprop.py
sonnet/src/optimizers/rmsprop_test.py
sonnet/src/optimizers/sgd.py
sonnet/src/optimizers/sgd_test.py