coverage ~=7.8.2
pytest ==8.3.5
pytest-benchmark ==5.1.0
pytest-timeout ==2.4.0
pytest-cov ==6.1.1
pytest-xdist ==3.7.0
pytest-random-order ==1.1.1
pytest-timestamper ==0.0.10
graphviz ==0.20.3
fdm ==0.5.0
expecttest ==0.3.0  # for test_ddp.py
hypothesis ~=6.133.0  # for test_ddp.py
numpy
einops  # for test_einops.py
litgpt==0.4.11  # for the model definition in tests and benchmarks
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.42.0  # fixed version!
transformers==4.52.4 # for test_networks.py

# Installs JAX on Linux and MacOS
jaxlib; sys_platform == 'linux' or sys_platform == 'darwin'  # required for jax, see https://github.com/google/jax#installation
jax; sys_platform == 'linux' or sys_platform == 'darwin'  # for test_ops.py

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