rikai>=0.1.1
torch
torchvision

[dev]
black
isort
pytest
