3 Commits

Author	SHA1	Message	Date
Tri Dao	bdcae547c7	[LayerNorm] Don't exit early in the backward pass (fix #781)	2024-01-22 22:40:06 -08:00
Tri Dao	c9861a032d	[LayerNorm] Initialize mean and rstd tensor using x.device	2024-01-09 16:30:31 -08:00
Tri Dao	f5b308e258	[LayerNorm] Rename layernorm.py -> layer_norm.py	2024-01-05 00:21:03 -08:00