torch>=2.2
transformers>=4.44
numpy>=1.24
pyyaml>=6.0
decord>=0.6.0
fvcore>=0.1.5
thop>=0.1.1
torchao>=0.5.0
onnx>=1.16
onnxruntime>=1.18

[bnb]
bitsandbytes>=0.43.0

[dev]
pytest>=8.0
matplotlib>=3.8
seaborn>=0.13

[trt]
tensorrt

[viz]
matplotlib>=3.8
seaborn>=0.13
