requirements.txt

accelerate>=0.23.0
beartype>=0.14.1
better-abc>=0.0.3
datasets>=2.7.1
einops>=0.6.0
fancy-einsum>=0.0.3
jaxtyping>=0.2.11
numpy>=1.24  # Assuming Python 3.9+
pandas>=1.1.5
rich>=12.6.0
sentencepiece
torch>=1.10
tqdm>=4.64.1
transformers>=4.41,<4.42  # Assuming Python 3.9+
typing-extensions
wandb>=0.13.5
typeguard>=4.2
pytest>=7.2.0  # From dev dependencies, needed by transformer_lens
gradio==4.19.2  # Keeping your existing Gradio requirement
transformer_lens==1.13.0