flax

0edd5189 - Clarifies ordering of dimensions in Convs