GitHub repository: szymonmaszke/torchlayers
Primary language: Python
Description: Shape and dimension inference (Keras-like) for PyTorch layers and neural networks