coverage ==7.4.4
pytest ==8.1.1
pytest-timeout ==2.2.0
pytest-cov ==4.1.0
pytest-xdist ==3.5.0
pytest-random-order ==1.1.1
pytest-timestamper ==0.0.10
graphviz ==0.20.1
fdm ==0.4.1
expecttest ==0.2.1  # for test_ddp.py
hypothesis ==6.100.0  # for test_ddp.py
numpy  # for test_ops.py
einops  # for test_einops.py
litgpt==0.3.0  # 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

# 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
