LICENSE
README.md
pyproject.toml
src/modelbatch/__init__.py
src/modelbatch/callbacks.py
src/modelbatch/core.py
src/modelbatch/data.py
src/modelbatch/huggingface_integration.py
src/modelbatch/logger.py
src/modelbatch/optimizer.py
src/modelbatch/optuna_integration.py
src/modelbatch/utils.py
src/modelbatch.egg-info/PKG-INFO
src/modelbatch.egg-info/SOURCES.txt
src/modelbatch.egg-info/dependency_links.txt
src/modelbatch.egg-info/requires.txt
src/modelbatch.egg-info/top_level.txt
tests/test_amp_optimizer.py
tests/test_compile.py
tests/test_consistency.py
tests/test_core.py
tests/test_custom_models.py
tests/test_huggingface_integration.py
tests/test_logger.py
tests/test_models.py
tests/test_optimizer.py
tests/test_optuna_integration.py