| # --------- pytorch --------- # | |
| torch>=2.0.0 | |
| torchvision>=0.15.0 | |
| lightning>=2.0.0 | |
| torchmetrics>=0.11.4 | |
| # --------- hydra --------- # | |
| hydra-core==1.3.2 | |
| hydra-colorlog==1.2.0 | |
| hydra-optuna-sweeper==1.2.0 | |
| # --------- loggers --------- # | |
| wandb | |
| # neptune-client | |
| # mlflow | |
| # comet-ml | |
| # aim>=3.16.2 # no lower than 3.16.2, see https://github.com/aimhubio/aim/issues/2550 | |
| # --------- others --------- # | |
| rootutils # standardizing the project root setup | |
| pre-commit # hooks for applying linters on commit | |
| rich # beautiful text formatting in terminal | |
| pytest # tests | |
| # sh # for running bash commands in some tests (linux/macos only) | |