Fixes `random_flax_module` with `flax.linen.BatchNorm` #3390

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
lint (3.9)	3m 57s
test-inference (3.9)	28m 11s
examples (3.9)	10m 29s
test-modeling (3.9)	23m 19s
	1h 5m 56s