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]))
Install Sonnet by running:
$ pip install tensorflow-gpu $ pip install "dm-sonnet>=2.0.0b0" --pre
Sonnet is licensed under the Apache 2.0 License.