Deepmind Python Packages

pysc2

StarCraft II Learning Environment

nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

disco-torch

PyTorch port of DeepMind's Disco103 meta-learned RL update rule

reaver

Reaver: Modular Deep Reinforcement Learning Framework. Focused on StarCraft II. Supports Gym, Atari, and MuJoCo.

nalu

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

turbokv

First open-source implementation of TurboQuant (arXiv 2504.19874) — 4-7x LLM KV cache compression. pip install turbokv

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment

alphagenome-viewer

Web interface for Google DeepMind's AlphaGenome genomic prediction model

turboquant-impl

First open-source implementation of TurboQuant (arXiv 2504.19874) — 4-7x LLM KV cache compression. pip install turbokv