flax

69b37323 - changed resnet v1 to v1.5