torch>=1.6
labml>=0.4.52
labml-helpers
torchvision
numpy>=1.16.3
matplotlib>=3.0.3
einops>=0.3.0
gym[atari]
opencv-python