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