torch
fastai
fasterai
torch-pruning