Migrate to the latest flax.nnx module from the legacy flax.linen module in Character-level Transformer on Tiny Shakespeare example.#1709
Open
rajasekharporeddy wants to merge 1 commit into