# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu tqdm datasets jinja2 transformers wandb # litgpt[all] litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git litdata grokadamw # bitsandbytes # pyzstd # zstd nvfuser-cu121-torch24 nvidia-cudnn-frontend lightning-thunder