soft entropy estimation

Code for computing discrete approximations to continuous entropy through soft-binning.

Code for ICLR 2023 Paper on Variation in Latent Representations

Fast Differentiable Sorting and Ranking

Coming soon...

Vector (and Scalar) Quantization, in Pytorch

Trax — Deep Learning with Clear Code and Speed

Simple language-driven navigation tasks for studying compositional learning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)

EGG: Emergence of lanGuage in Games

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.