tensorflow-probability
gast
torch
torchvision
smart_open