torch
compressai
numpy
