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

Commits
  • Implement BoxedParam as flax.core.AxisMetadata.
    a-googler committed 3 years ago
Loading