* Bug fix for non-affine layer-norm + add backward unit test * clean up tests and add tests for a large batch
Attach a file by drag & drop or click to upload
