flax
mnist: tweak demo call
#1113
Merged

Loading