polars
pyarrow
nested_ragged_tensors
loguru
numpy
ml-mixins
torch
omegaconf
hydra-core
lightning
x-transformers
rich
hydra-colorlog
hydra-optuna-sweeper
wandb
torchmetrics
torchvision
transformers
ray[tune]

[dev]
pre-commit
notebook
pdbpp

[mamba]
mamba-ssm
causal-conv1d>=1.4.0

[tests]
pytest
pytest-cov[toml]
rootutils
pytest-sugar
pytest-instafail
pytest-xdist
sh
MEDS-transforms==0.0.6
