numpyneuron/__init__.py

from .loss import *
from .activation import *
from .nn import *

ACTIVATIONS: dict[str, Activation] = {
    "Relu": Relu(),
    "Sigmoid": Sigmoid(),
    "TanH": TanH(),
    "SoftMax": SoftMax(),
}