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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
lint (3.9)	3m 52s
test-modeling	0s
examples	0s
test-inference	0s
	3m 52s