open-nudge/torchlayers
A PyTorch extension providing Keras-style automatic shape and dimensionality inference for neural network layers.

Velocity · 7d
+0.2
★ / day
Trend
→steady
star history
torchlayers is a library that extends PyTorch with automatic shape inference for nn layers, removing the need for manual dimension specification. It implements current state-of-the-art architecture building blocks such as those used in EfficientNet. The library functions similarly to Keras in terms of layer construction while remaining fully compatible with native PyTorch.