requirements.txt

# --------- pytorch --------- #
torch>=2.0.0
torchvision>=0.15.0
lightning>=2.0.0
torchmetrics>=0.11.4

# --------- hydra --------- #
hydra-core>=1.0
hydra-colorlog
hydra-optuna-sweeper

# --------- 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)

torchdiffeq
torchcfm

black
flake8
isort
einops
nflows
normflows
git+https://github.com/VincentStimper/resampled-base-flows.git # larsflow
git+https://github.com/noegroup/bgflow.git
git+https://github.com/jarridrb/fab-torch.git
scikit-learn
scipy