polars~=1.26.0
pyarrow
hydra-core
numpy
meds~=0.4.0
filelock
pytest
meds_testing_helpers~=0.3.0
MEDS_transforms~=0.5.0
meds-torch-data[lightning]~=0.6.0
transformers
torch
torchmetrics
lightning~=2.5.1

[dev]
pre-commit<4
ruff

[flash_attn]
flash-attn

[mlflow]
mlflow

[tests]
pytest-cov
pretty_print_directory>=0.1.2

[wandb]
wandb
