flax
[NVIDIA] Rename fp8 custom dtype to `fp32_max_grad`
#3984

Merged

[NVIDIA] Rename fp8 custom dtype to `fp32_max_grad` #3984

copybara-service merged 2 commits into google:main from kaixih:rename_fp8_custom_dtype

IvyZX approved these changes on 2024-07-02

Rename the new dtype

16f28666

use fmax32

798cfe7b

kaixih force pushed to 798cfe7b 1 year ago

IvyZX added pull ready

copybara-service merged b3785dc7 into main 1 year ago

Reviewers

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone