.gitignore
.pre-commit-config.yaml
.python-version
LICENSE
README.rst
pyproject.toml
sumi.toml
uv.lock
.github/workflows/docs.yml
.github/workflows/publish.yml
.github/workflows/tests.yml
data/logo/logo-background.png
data/logo/logo-transparent.png
docs/Makefile
docs/make.bat
docs/requirements.txt
docs/source/conf.py
docs/source/getting-started.rst
docs/source/index.rst
docs/source/optuna.rst
docs/source/plugins.rst
docs/source/quick-ref.rst
docs/source/tutorial.rst
examples/classifier_tutorial.py
examples/optuna_tutorial.py
src/helios/__init__.py
src/helios/_version.py
src/helios/chkpt_migrator.py
src/helios/onnx.py
src/helios/py.typed
src/helios/trainer.py
src/helios/core/__init__.py
src/helios/core/cuda.py
src/helios/core/distributed.py
src/helios/core/logging.py
src/helios/core/rng.py
src/helios/core/utils.py
src/helios/data/__init__.py
src/helios/data/datamodule.py
src/helios/data/functional.py
src/helios/data/samplers.py
src/helios/data/transforms.py
src/helios/losses/__init__.py
src/helios/losses/utils.py
src/helios/losses/weighted_loss.py
src/helios/metrics/__init__.py
src/helios/metrics/functional.py
src/helios/metrics/metrics.py
src/helios/model/__init__.py
src/helios/model/model.py
src/helios/model/utils.py
src/helios/nn/__init__.py
src/helios/nn/swa_utils.py
src/helios/nn/utils.py
src/helios/nn/layers/__init__.py
src/helios/nn/layers/pool.py
src/helios/optim/__init__.py
src/helios/optim/utils.py
src/helios/plugins/__init__.py
src/helios/plugins/plugin.py
src/helios/plugins/optuna/__init__.py
src/helios/plugins/optuna/plugin.py
src/helios/plugins/optuna/utils.py
src/helios/scheduler/__init__.py
src/helios/scheduler/schedulers.py
src/helios/scheduler/utils.py
src/helios_ml.egg-info/PKG-INFO
src/helios_ml.egg-info/SOURCES.txt
src/helios_ml.egg-info/dependency_links.txt
src/helios_ml.egg-info/requires.txt
src/helios_ml.egg-info/top_level.txt
test/__init__.py
test/conftest.py
test/test_chkpt_migrator.py
test/test_core.py
test/test_data.py
test/test_losses.py
test/test_metrics.py
test/test_model.py
test/test_nn.py
test/test_onnx.py
test/test_optim.py
test/test_optuna.py
test/test_plugins.py
test/test_scheduler.py
test/test_trainer.py
test/registry_test/__init__.py
test/registry_test/extra.py
test/registry_test/foo.py
test/registry_test/func_registry.py
test/registry_test/py.typed
test/registry_test/nested/__init__.py
test/registry_test/nested/bar.py
test/registry_test/nested/extra.py