issues
search
google
/
flax
Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.79k
stars
609
forks
source link
updated nnx attention and nn docstrings
#3971
Closed
chiamp
closed
1 month ago
chiamp
commented
1 month ago
updated nnx attention and nn docstrings
updated nnx attention and nn docstrings