polars<2,>=1.32
pyarrow
hydra-core
numpy
meds~=0.4.0
filelock
pytest
meds_testing_helpers~=0.3.0
MEDS-transforms~=0.5.2
meds-torch-data[lightning]~=0.7.0
transformers
torch
torchmetrics
lightning~=2.5.1
MEDS_trajectory_evaluation~=0.0.3

[mlflow]
mlflow
psutil
pynvml

[wandb]
wandb
