← 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
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is open-nudge/torchlayers?
A PyTorch extension providing Keras-style automatic shape and dimensionality inference for neural network layers.
Is torchlayers open source?
Yes — open-nudge/torchlayers is open source, released under the MIT license.
What language is torchlayers written in?
open-nudge/torchlayers is primarily written in Python.
How popular is torchlayers?
open-nudge/torchlayers has 570 stars on GitHub.
Where can I find torchlayers?
open-nudge/torchlayers is on GitHub at https://github.com/open-nudge/torchlayers.

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