numpy
six
tensorflow-probability

[tensorflow]
tensorflow>=1.14

[tensorflow_gpu]
tensorflow-gpu>=1.14

[tests]
dm-sonnet
pytest
