NLP environment install rational cuda git clone https://github.com/Adamdad/rational_kat_cu.git cd rational_kat_cu pip install -e . dataset python3 nanoGPT/data/openwebtext/prepare.py run sbatch run.sh