| # conda install cuda=12.4 cuda-toolkit=12.4 cuda-compiler=12.4 cuda-cudart=12.4 | |
| # 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 | |