CompressAI provides custom operations, layers, and pre-trained models for deep learning based image and video compression. It includes a partial port of the TensorFlow compression library and evaluation scripts to benchmark learned compression models against classical codecs. The library supports multi-GPU training and is designed for compression research.