[JAX] Support layernorm/rmsnorm sm_margin control through environment variable#520

Merged

denera merged 1 commit intoNVIDIA:mainfrom zlsh80826:rewang/support-ln-sm-margin

Nov 30, 2023

+55-25

Commits on Nov 27, 2023

Support layernorm sm_margin through environment variables
zlsh80826
committed