RNN-Based Text Generation Project (JAX, NLP)
Date:
Developed a character-level language model using JAX, trained on the Penn Treebank dataset.
Implemented autoregressive decoding (sampling, greedy, and beam search) entirely in vectorized JAX for efficient, pure-python loops-free generation.
(GitHub Link)