# Test utilities
pytest
unittest-xml-reporting
parameterized
packaging
transformers
hypothesis # Avoid test derandomization warning
sentencepiece # for gpt-fast tokenizer
expecttest

# For prototype features and benchmarks
bitsandbytes # needed for testing triton quant / dequant ops for 8-bit optimizers
matplotlib
pandas
fire  # QOL for commandline scripts
tabulate  # QOL for printing tables to stdout
tiktoken
blobfile
lm_eval
# sam
diskcache
pycocotools
tqdm
importlib_metadata

# Custom CUDA Extensions
ninja

# CPU kernels
cmake<4.0.0,>=3.19.0

# Linting
ruff==0.11.6
pre-commit