LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
pytagi/__init__.py
pytagi/__version.py
pytagi/cuda.py
pytagi/metric.py
pytagi/tagi_utils.py
pytagi/version.txt
pytagi.egg-info/PKG-INFO
pytagi.egg-info/SOURCES.txt
pytagi.egg-info/dependency_links.txt
pytagi.egg-info/top_level.txt
pytagi/nn/__init__.py
pytagi/nn/activation.py
pytagi/nn/base_layer.py
pytagi/nn/batch_norm.py
pytagi/nn/conv2d.py
pytagi/nn/convtranspose2d.py
pytagi/nn/data_struct.py
pytagi/nn/ddp.py
pytagi/nn/layer_block.py
pytagi/nn/layer_norm.py
pytagi/nn/linear.py
pytagi/nn/lstm.py
pytagi/nn/output_updater.py
pytagi/nn/pooling.py
pytagi/nn/resnet_block.py
pytagi/nn/sequential.py
pytagi/nn/slinear.py
pytagi/nn/slstm.py