numpy
tensorflow>=2.1.0
colr
srsly
pydantic>=1.0.0
click
wasabi
gym<=0.12.5
tensorflow_probability
typing-extensions>=3.7.4.1
keras-self-attention

[dev]
mypy
pytest
pytest-cov
black
flake8
snakeviz
tensorboard>=2.0.0
