sonnet

Sonnet 是一个基于 TensorFlow 2 构建的神经网络库，旨在为机器学习研究提供简单且可组合的抽象。其核心功能围绕 `snt.Module` 概念设计，模块可以持有参数、其他模块以及对用户输入应用函数的方法。Sonnet 提供了多种预定义模块（如 `snt.Linear`、`snt.Conv2D` 和 `snt.BatchNorm`）及一些预定义的模块网络（如 `snt.nets.MLP`），同时也鼓励用户自定义模块。该库不包含训练框架，允许用户完全自由地构建或采用第三方解决方案。Sonnet 适合用于各种目的的神经网络构建，包括有监督学习、无监督学习和强化学习等场景，并以其简洁易懂的设计而著称。