torch
torchvision
salesforce-lavis
openai