torch >=1.11.0
numpy >1.20.0, <2.0
pandas >=1.1.5
scikit-learn >=1.3.0
pytorch-lightning >=2.0.0, <2.5.0
omegaconf >=2.3.0
torchmetrics >=0.10.0, <1.7.0
tensorboard >2.2.0, !=2.5.0
protobuf >=3.20.0, <5.30.0
pytorch-tabnet ==4.1
PyYAML >=5.4, <6.1.0
# importlib-metadata <1,>=0.12
matplotlib >3.1
ipywidgets
einops >=0.6.0, <0.8.0
rich >=11.0.0
