flax
Implement BoxedParam as flax.core.AxisMetadata.
#2789
Open

Loading