torch
hydra-core
pytorch-lightning
einops
wandb
matplotlib