Efficient layer normalization GPU kernel for Tensorflow - View it on GitHub