coverage ~=7.9.1
pytest ==8.3.5
pytest-benchmark ==5.1.0
pytest-timeout ==2.4.0
pytest-cov ==6.2.1
pytest-xdist ==3.8.0
pytest-random-order ==1.2.0
pytest-timestamper ==0.0.10
graphviz ==0.21
fdm ==0.5.0
expecttest ==0.3.0  # for test_ddp.py
hypothesis ~=6.136.6  # for test_ddp.py
numpy
einops  # for test_einops.py
litgpt==0.5.0  # for the model definition in tests and benchmarks  # todo: need update to latest
absl-py # thunder/benchmarks/test_benchmark_litgpt.py
pandas # thunder/benchmarks/test_benchmark_litgpt.py
xlsxwriter # thunder/benchmarks/test_benchmark_litgpt.py
jsonargparse # thunder/benchmarks/benchmark_litgpt.py
bitsandbytes==0.47.0; 'arm' not in platform_machine and 'aarch' not in platform_machine
bitsandbytes>=0.42,<0.43; 'arm' in platform_machine or 'aarch' in platform_machine
transformers==4.52.4 # for test_networks.py
diffusers==0.34.0 # for test_networks.py
accelerate # for test_networks.py

asvdb @ git+https://github.com/rapidsai/asvdb.git
asv >=0.6.4
