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.78k
stars
610
forks
source link
Port gemma/layers to NNX
#4013
Closed
copybara-service[bot]
closed
1 week ago
copybara-service[bot]
commented
1 week ago
Port gemma/layers to NNX
Port gemma/layers to NNX