← all repositories

open-nudge/torchlayers

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

570 stars Python ML Frameworks
torchlayers
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.