Improved time to first gradient#151

Merged

ToucheSir merged 3 commits intoFluxML:masterfrom

theabhirath:conv_bn

May 1, 2022

Commits on Apr 27, 2022

Commits on Apr 29, 2022

Formatting, try to get all tests to pass
theabhirath
committed