flax
Explicitly mark the lazy-rng with legacy_config, and all new configs should use flax.config.
#2895

Merged

Commits

Explicitly mark the lazy-rng with legacy_config, and all new configs should use flax.config.

a-googler committed 3 years ago

Loading