rnn from scratch! very lightweight JAX rnn for educational purposes open me in colab: (and if you do, download the one-fish-two-fish.txt file to re-upload to your colab instance) or, if local, to install reqs: python3 -m pip install -r requirements.txt