.gitignore
.pre-commit-config.yaml
CITATION.cff
LICENSE
README.md
conftest.py
pyproject.toml
uv.lock
.github/actions/setup/action.yaml
.github/workflows/code-quality-main.yaml
.github/workflows/code-quality-pr.yaml
.github/workflows/python-build.yaml
.github/workflows/tests.yaml
src/MEDS_EIC_AR/__init__.py
src/MEDS_EIC_AR/__main__.py
src/MEDS_EIC_AR/utils.py
src/MEDS_EIC_AR.egg-info/PKG-INFO
src/MEDS_EIC_AR.egg-info/SOURCES.txt
src/MEDS_EIC_AR.egg-info/dependency_links.txt
src/MEDS_EIC_AR.egg-info/entry_points.txt
src/MEDS_EIC_AR.egg-info/requires.txt
src/MEDS_EIC_AR.egg-info/top_level.txt
src/MEDS_EIC_AR/configs/README.md
src/MEDS_EIC_AR/configs/__init__.py
src/MEDS_EIC_AR/configs/_demo_generate_trajectories.yaml
src/MEDS_EIC_AR/configs/_demo_pretrain.yaml
src/MEDS_EIC_AR/configs/_generate_trajectories.yaml
src/MEDS_EIC_AR/configs/_pretrain.yaml
src/MEDS_EIC_AR/configs/datamodule/__init__.py
src/MEDS_EIC_AR/configs/datamodule/default.yaml
src/MEDS_EIC_AR/configs/datamodule/generate_trajectories.yaml
src/MEDS_EIC_AR/configs/datamodule/pretrain.yaml
src/MEDS_EIC_AR/configs/inference/default.yaml
src/MEDS_EIC_AR/configs/inference/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/large.yaml
src/MEDS_EIC_AR/configs/lightning_module/medium.yaml
src/MEDS_EIC_AR/configs/lightning_module/micro.yaml
src/MEDS_EIC_AR/configs/lightning_module/small.yaml
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/LR_scheduler/get_cosine_schedule_with_warmup.yaml
src/MEDS_EIC_AR/configs/lightning_module/metrics/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/metrics/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/model/default.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/demo.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/large.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/medium.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/micro.yaml
src/MEDS_EIC_AR/configs/lightning_module/model/small.yaml
src/MEDS_EIC_AR/configs/lightning_module/optimizer/__init__.py
src/MEDS_EIC_AR/configs/lightning_module/optimizer/adamw.yaml
src/MEDS_EIC_AR/configs/trainer/__init__.py
src/MEDS_EIC_AR/configs/trainer/default.yaml
src/MEDS_EIC_AR/configs/trainer/demo.yaml
src/MEDS_EIC_AR/configs/trainer/demo_generate.yaml
src/MEDS_EIC_AR/configs/trainer/generate.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/__init__.py
src/MEDS_EIC_AR/configs/trainer/callbacks/default.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/early_stopping.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/generation.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/generation_speed_logger.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/learning_rate_monitor.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/model_checkpoint.yaml
src/MEDS_EIC_AR/configs/trainer/callbacks/save_logger_run_ids.yaml
src/MEDS_EIC_AR/configs/trainer/logger/__init__.py
src/MEDS_EIC_AR/configs/trainer/logger/csv.yaml
src/MEDS_EIC_AR/configs/trainer/logger/mlflow.yaml
src/MEDS_EIC_AR/configs/trainer/logger/wandb.yaml
src/MEDS_EIC_AR/generation/__init__.py
src/MEDS_EIC_AR/generation/finalize.py
src/MEDS_EIC_AR/generation/repeated_dataset.py
src/MEDS_EIC_AR/generation/utils.py
src/MEDS_EIC_AR/model/README.md
src/MEDS_EIC_AR/model/__init__.py
src/MEDS_EIC_AR/model/model.py
src/MEDS_EIC_AR/model/backends/__init__.py
src/MEDS_EIC_AR/model/backends/base.py
src/MEDS_EIC_AR/model/backends/hf.py
src/MEDS_EIC_AR/preprocessing/README.md
src/MEDS_EIC_AR/preprocessing/__init__.py
src/MEDS_EIC_AR/preprocessing/__main__.py
src/MEDS_EIC_AR/preprocessing/configs/_data.yaml
src/MEDS_EIC_AR/preprocessing/configs/_process_data.yaml
src/MEDS_EIC_AR/preprocessing/configs/_reshard_data.yaml
src/MEDS_EIC_AR/training/README.md
src/MEDS_EIC_AR/training/__init__.py
src/MEDS_EIC_AR/training/callbacks.py
src/MEDS_EIC_AR/training/files.py
src/MEDS_EIC_AR/training/metrics.py
src/MEDS_EIC_AR/training/module.py
tests/__init__.py
tests/test_environment_snapshot.py
tests/test_format_trajectories_overflow_cli.py
tests/test_generate_trajectories.py
tests/test_generation_backend.py
tests/test_logger_helpers.py
tests/test_logger_helpers_integration.py
tests/test_logger_run_id_persistence.py
tests/test_pretrain.py
tests/test_process_data.py
tests/grammar/__init__.py
tests/grammar/_grammar.py
tests/grammar/_meds.py
tests/grammar/conftest.py
tests/grammar/test_cli.py
tests/grammar/test_in_process.py