dm-sonnet
Sonnet is a library built on top of TensorFlow 2 designed to provide simple, composable abstractions for machine learning research.
Sonnet is a library built on top of TensorFlow 2 designed to provide simple, composable abstractions for machine learning research.
To install this package, run one of the following:
Sonnet has been designed and built by researchers at DeepMind. It can be used to construct neural networks for many different purposes (un/supervised learning, reinforcement learning, ...). More specifically, Sonnet provides a simple but powerful programming model centered around a single concept: snt.Module. Modules can hold references to parameters, other modules and methods that apply some function on the user input. Sonnet ships with many predefined modules (e.g. snt.Linear, snt.Conv2D, snt.BatchNorm) and some predefined networks of modules (e.g. snt.nets.MLP) but users are also encouraged to build their own modules.
Summary
Sonnet is a library built on top of TensorFlow 2 designed to provide simple, composable abstractions for machine learning research.
Last Updated
Jan 3, 2024 at 18:40
License
Apache-2.0
Total Downloads
48.0K
Supported Platforms
Unsupported Platforms
GitHub Repository
https://github.com/deepmind/sonnetDocumentation
https://sonnet.readthedocs.io/en/latest/