sonnet

收录理由

Sonnet 出自 DeepMind 研究团队之手，他们日常就用它跑自己的实验，所以这个库带着真实研究项目的分量，而不是一个包装精美的演示品。整个库围绕 snt.Module 这一个核心概念展开，开箱即用的组件包括 Linear、Conv2D、BatchNorm，以及预构建的 MLP 网络。但它并不附带训练循环——Sonnet 刻意保持中立，模块之间相互独立、彼此解耦，你可以自由拼装自定义架构，而不用跟框架较劲。对于做强化学习、分布式训练，或者任何希望看清并掌控每一层具体行为的团队来说，这种可读性和控制力通常比一个什么都替你包办的全家桶式框架更实用。