GitHub repository: EtienneChollet/cassetta
Description: A PyTorch toolbox