eval/evaluation
nn/activations
nn/backpropagation
nn/forward
nn/layers
nn/losses
nn/model
nn/parameters
optim/optimizers
utils/dataloader
utils/process_tensor
utils/transforms
viz/visualization
