torchtuples (conda): Training neural networks in PyTorch
Keywords: deep-learning|machine-learning|neural-network|python|pytorch