Package                  Version     Editable project location
------------------------ ----------- -----------------------------
assertpy                 1.1
Brotli                   1.1.0
causal-conv1d            1.4.0
certifi                  2024.7.4
cffi                     1.16.0
charset-normalizer       3.3.2
cupy                     13.2.0
einops                   0.8.0
fastargs                 1.2.0
fastrlock                0.8.2
ffcv                     1.0.1       /mnt/raid/data/MatMamba/ffcv
filelock                 3.15.4
fsspec                   2024.2.0
gmpy2                    2.1.5
h2                       4.1.0
hpack                    4.0.0
huggingface-hub          0.24.2
hyperframe               6.0.1
idna                     3.7
Jinja2                   3.1.4
lightning-utilities      0.11.6
llvmlite                 0.43.0
mamba_ssm                2.2.2       /mnt/raid/data/MatMamba/mamba
MarkupSafe               2.1.5
matmamba                 0.1         /mnt/raid/data/MatMamba
mpmath                   1.3.0
networkx                 3.2.1
ninja                    1.11.1.1
numba                    0.60.0
numpy                    2.0.0
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.1.105
nvidia-nvtx-cu12         12.1.105
opencv-python            4.10.0
opencv-python-headless   4.10.0
packaging                24.1
pillow                   10.4.0
pip                      24.0
psutil                   6.0.0
pycparser                2.22
PySocks                  1.7.1
pytorch-pfn-extras       0.7.7
PyYAML                   6.0.1
regex                    2024.7.24
requests                 2.32.3
safetensors              0.4.3
setuptools               71.0.4
sympy                    1.13.0
terminaltables           3.1.10
tiktoken                 0.7.0
timm                     1.0.7
tokenizers               0.19.1
torch                    2.4.0+cu121
torchaudio               2.4.0+cu121
torchmetrics             1.4.0.post0
torchvision              0.19.0
tqdm                     4.66.4
transformers             4.43.2
triton                   3.0.0
typing_extensions        4.12.2
urllib3                  2.2.2
wheel                    0.43.0
zstandard                0.23.0
