flax
Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm.
#3510
Merged

Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm. #3510

ghost merged 1 commit into google:main from normalization_mask
carlosgmartin
chiamp
carlosgmartin Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm.
59c6c30c
carlosgmartin
chiamp chiamp added pull ready
copybara-service copybara-service closed this 2 years ago
carlosgmartin carlosgmartin deleted the normalization_mask branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone