einops
numpy
opencv-python
scipy
lightning
tqdm
albumentations
timm
python-dotenv
git+https://github.com/google/dopamine.git
tensorflow
gin

umap-learn
matplotlib
seaborn
bokeh 
holoviews 
scikit-image 
colorcet

atari-py
gym[atari]

wandb
