pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118
pip install -e causal-conv1d
pip install -e mamba