dataclasses-json>=0.6.3
torch>=2.1.2
