flax
Improve nn.scan docs
#2839
Merged

Loading