polars~=1.35
pyarrow
hydra-core
numpy
meds~=0.4.0
filelock
pytest
meds_testing_helpers~=0.3.0
MEDS-transforms~=0.6.7
meds-torch-data[lightning]~=0.9.0
transformers>=4.56
torch
torchmetrics
lightning~=2.5.1
MEDS_trajectory_evaluation~=0.0.6

[mlflow]
mlflow
psutil
pynvml

[wandb]
wandb
