#
nnx
Here are 6 public repositories matching this topic...
-
Updated
Sep 23, 2024 - Jupyter Notebook
A JAX / NNX implementation of a linear GRU for language modeling
-
Updated
Oct 31, 2024 - Jupyter Notebook
Improve this page
Add a description, image, and links to the nnx topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the nnx topic, visit your repo's landing page and select "manage topics."