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
6k
stars
633
forks
source link
Port gemma/modules to NNX
#4014
Closed
copybara-service[bot]
closed
3 months ago
copybara-service[bot]
commented
3 months ago
Port gemma/modules to NNX
Port gemma/modules to NNX