candle-pyo3/py_src/candle/nn/__init__.py


1
2
3
4
5

from .module import Module
from .container import Sequential, ModuleList, ModuleDict
from .sparse import Embedding
from .normalization import LayerNorm
from .linear import Linear