nnx

github.com ↗

A completely unstable and experimental package that extends Torch's builtin nn library.