Sonnet Documentation

Sonnet is a library built on top of TensorFlow designed to provide simple, composable abstractions for machine learning research.

import sonnet as snt
import tensorflow as tf

mlp = snt.nets.MLP([1024, 1024, 10])
logits = mlp(tf.ones([8, 28 * 28]))

Installation

Install Sonnet by running:

$ pip install tensorflow
$ pip install dm-sonnet

Guides

• Modules

Package Reference

• Base
• Linear modules
• Convolutional modules
• Normalization modules
• Recurrent modules
• Batch
• Embedding modules
• Optimizers
• Initializers
• Regularizers
• Paddings
• Distribution
• Metrics
• Nets
• Mixed Precision
• References

Contribute

• Issue tracker: https://github.com/deepmind/sonnet/issues

• Source code: https://github.com/deepmind/sonnet/tree/v2

Support

If you are having issues, please let us know by filing an issue on our issue tracker.

License

Sonnet is licensed under the Apache 2.0 License.

Indices and tables

• Index

• Module Index