Move layer norm to new schedule (#2590)
* Move layer norm to new schedule
* Pre-commit fixes
* fix comments
* format fixes
* Merge unrolls
* format fixes
* camelCase
* format fixes
* revert unwanted file
* move pow2 function
* format fixes
Co-authored-by: Connor Holmes <connorholmes@microsoft.com>
