LICENSE
NOTICE
README.md
pyproject.toml
src/castor_gw/__init__.py
src/castor_gw/checkpoints.py
src/castor_gw/evaluation.py
src/castor_gw/preprocessing.py
src/castor_gw/reproducibility.py
src/castor_gw.egg-info/PKG-INFO
src/castor_gw.egg-info/SOURCES.txt
src/castor_gw.egg-info/dependency_links.txt
src/castor_gw.egg-info/entry_points.txt
src/castor_gw.egg-info/requires.txt
src/castor_gw.egg-info/top_level.txt
src/castor_gw/cli/__init__.py
src/castor_gw/cli/evaluate.py
src/castor_gw/cli/train.py
src/castor_gw/data/__init__.py
src/castor_gw/data/batches.py
src/castor_gw/data/loaders.py
src/castor_gw/data/ml4gw.py
src/castor_gw/data/pregenerated.py
src/castor_gw/models/__init__.py
src/castor_gw/models/castor.py
src/castor_gw/models/statistics.py
src/castor_gw/training/__init__.py
src/castor_gw/training/curriculum.py
src/castor_gw/training/ema.py
src/castor_gw/training/losses.py
src/castor_gw/training/trainer.py
tests/test_checkpoint_resume.py
tests/test_checkpoint_weights.py
tests/test_ema.py
tests/test_evaluation.py
tests/test_import.py
tests/test_losses.py
tests/test_ml4gw_provider.py
tests/test_model.py
tests/test_pregenerated_data.py
tests/test_preprocessing.py
tests/test_reproducibility.py
tests/test_slurm_evaluation.py
tests/test_statistics.py
tests/test_train_cli.py
tests/test_training_pipeline.py