# Test utilities
pytest==7.4.0
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

# Custom CUDA Extensions
ninja

# Linting
ruff
pre-commit