Code to reproduce results of the paper Rethinking Neural Operations for Diverse Tasks. The chrysalis folder contains the core code and applications, running which requires PyTorch and Butterfly, in addition to the requirements in chrysalis/requirements.txt.
If extending or applying XD-operations, we recommend using the relax software instead.
Install instructions below; a Dockerfile is also provided.
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch-nightly
git clone --single-branch --branch pt1.8 https://github.com/HazyResearch/butterfly.git
cd butterfly && export FORCE_CUDA="1" && python setup.py install