requirements.txt

accelerate==1.0.1
beartype==0.19.0
certifi==2024.12.14
charset-normalizer==3.4.1
classifier-free-guidance-pytorch==0.7.1
contourpy==1.1.1
cycler==0.12.1
einops==0.8.0
einx==0.3.0
ema-pytorch==0.7.7
environs==11.2.1
filelock==3.16.1
fonttools==4.55.3
frozendict==2.4.6
fsspec==2024.12.0
ftfy==6.2.3
gateloop-transformer==0.2.5
google-pasta==0.2.0
huggingface-hub==0.27.1
idna==3.10
importlib-metadata==8.5.0
importlib-resources==6.4.5
jaxtyping==0.2.19
jinja2==3.1.5
keras==2.13.1
kiwisolver==1.4.7
local-attention==1.9.15
MarkupSafe==2.1.5
marshmallow==3.22.0
matplotlib==3.7.5
meshgpt-pytorch==1.5.12
mpmath==1.3.0
networkx==3.1
numpy==1.24.3
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.6.85
nvidia-nvtx-cu12==12.1.105
open-clip-torch==2.30.0
packaging==24.2
pillow==10.4.0
psutil==6.1.1
pyparsing==3.1.4
python-dateutil==2.9.0.post0
python-dotenv==1.0.1
pytorch-custom-utils==0.0.21
pytorch-warmup==0.2.0
PyYAML==6.0.2
regex==2024.11.6
requests==2.32.3
rotary-embedding-torch==0.8.6
safetensors==0.5.2
sentencepiece==0.2.0
six==1.17.0
sympy==1.13.3
taylor-series-linear-attention==0.1.12
tensorflow-io-gcs-filesystem==0.34.0
timm==1.0.13
tokenizers==0.20.3
torch==2.4.1
torch-geometric==2.6.1
torchvision==0.19.1
tqdm==4.67.1
transformers==4.46.3
trimesh==4.5.3
triton==3.0.0
typeguard==4.4.0
typing-extensions==4.12.2
urllib3==2.2.3
vector-quantize-pytorch==1.21.1
wcwidth==0.2.13
x-transformers==1.30.23
zipp==3.20.2