| # 'torch' must be installed separately first, using the command | |
| # from the README.md to match your specific CUDA version. | |
| torchmetrics | |
| triton==3.2.0 | |
| numpy | |
| pandas | |
| matplotlib | |
| flash-linear-attention @ git+https://github.com/fla-org/flash-linear-attention@main | |
| scikit-learn | |
| gluonts | |
| notebook | |
| datasets | |
| ujson | |
| pyyaml | |
| wandb | |
| build | |
| pre-commit | |
| ruff | |
| mypy | |
| commitizen | |
| black | |
| cupy-cuda12x | |
| statsmodels | |
| pyo # Requires portaudio |