export PYTHONPATH=$PWD:$PYTHONPATH
export TORCH_EXTENSIONS_DIR=$PWD/tmp
export TORCH_HOME="$PWD/cache"
