google-deepmind/sonnet

Health v2 · maintenance only

Adoption not part of the score

9961 stars · 1303 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

Sonnet is a Python library built on TensorFlow 2 that provides simple, composable abstractions (snt.Module) for constructing neural networks in machine learning research. Developed by DeepMind, it ships with predefined modules like Linear, Conv2D, and MLP while remaining unopinionated about training workflows.

Use cases

• build neural networks for ML research in TensorFlow
• define MLPs and CNNs with composable modules
• prototype supervised and reinforcement learning models
• run distributed training on CIFAR-10 with snt.distribute
• train a GAN on MNIST in a Colab notebook

When to choose

• you use TensorFlow 2 and want clean, composable module abstractions
• you're doing ML research and prefer building your own training loop
• you want DeepMind's battle-tested neural network building blocks

When to avoid

• you want a full training framework with built-in trainers
• your project is PyTorch- or JAX-based
• you need a high-level Keras-style end-to-end API

Facets

library · maturity maintenance

machine-learning deep-learning machine-learning deep-learning artificial-intelligence python tensorflow neural-networks research deepmind modules

Member repositories

For agents

markdown · JSON · MCP: product_card(name="google-deepmind/sonnet")