| torch>=2.0.0 | |
| torchvision>=0.15.0 | |
| numpy>=1.24.0 | |
| tqdm>=4.65.0 | |
| pillow>=9.5.0 | |
| # Optional but recommended | |
| accelerate>=0.20.0 | |
| transformers>=4.30.0 | |
| einops>=0.6.1 | |
| wandb>=0.15.0 | |
| # For video I/O | |
| decord>=0.6.0 | |
| opencv-python>=4.7.0 | |
| imageio>=2.31.0 | |
| imageio-ffmpeg>=0.4.8 | |
| # Development | |
| pytest>=7.3.0 | |
| black>=23.3.0 | |
| flake8>=6.0.0 | |