README.md
pyproject.toml
tests/test_core.py
tests/test_embedding.py
universal_model/__init__.py
universal_model.egg-info/PKG-INFO
universal_model.egg-info/SOURCES.txt
universal_model.egg-info/dependency_links.txt
universal_model.egg-info/requires.txt
universal_model.egg-info/top_level.txt
universal_model/checkpoint/__init__.py
universal_model/checkpoint/checkpoint.py
universal_model/core/__init__.py
universal_model/core/module.py
universal_model/core/parameter.py
universal_model/core/tensor_ops.py
universal_model/data/__init__.py
universal_model/data/dataloader.py
universal_model/data/dataset.py
universal_model/data/split.py
universal_model/data/tensor_dataset.py
universal_model/evaluation/__init__.py
universal_model/evaluation/evaluator.py
universal_model/evaluation/metrics.py
universal_model/init/__init__.py
universal_model/init/initializers.py
universal_model/layers/__init__.py
universal_model/layers/activation.py
universal_model/layers/embedding.py
universal_model/layers/linear.py
universal_model/layers/sequential.py
universal_model/losses/__init__.py
universal_model/losses/mse.py
universal_model/models/__init__.py
universal_model/models/base.py
universal_model/models/config.py
universal_model/optim/__init__.py
universal_model/optim/optimizer.py
universal_model/optim/sgd.py
universal_model/state/__init__.py
universal_model/state/state.py
universal_model/training/__init__.py
universal_model/training/history.py
universal_model/training/trainer.py
universal_model/utils/__init__.py