flax
add usage note to BatchNorm docstring.
#1490
Merged

Loading