feat(nnx): add DyT (Dynamic Tanh) normalization module#5459

Ashutosh0x wants to merge 1 commit into

google:maingoogle/flax:mainfrom

Ashutosh0x:feat/dyt-normalizationAshutosh0x/flax:feat/dyt-normalizationCopy head branch name to clipboard