search

google / flax

Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
6.15k stars 649 forks source link

Update Flax NNX Glossary #4284

Closed 8bitmp3 closed 1 month ago

8bitmp3 commented 1 month ago

Updated certain terms, reworded, improved syntax and formatting.

Note: glossary.rst -> nnx_glossary.rst to address the automatic (unintentional) redirect to the old Flax Linen glossary on ReadTheDocs.

Preview: https://flax--4284.org.readthedocs.build/en/4284/nnx_glossary.html

8bitmp3 commented 1 month ago

Fixing the rendering in https://flax--4284.org.readthedocs.build/en/4284/nnx_glossary.html