# Core dependencies torch torchvision numpy pillow opencv-python transformers tokenizers scipy scikit-image pyyaml pandas tqdm webdataset accelerate xformers bitsandbytes jupyter matplotlib timm flash_attn