git clone -b v2.3.3 https://github.com/Dao-AILab/flash-attention && \
    cd flash-attention && MAX_JOBS=46 python setup.py install && \
    cd .. && \
    rm -rf flash-attention