flax
f39de8fa - Merge pull request #3374 from levskaya:fp8fix

Commit

2 years ago

Merge pull request #3374 from levskaya:fp8fix PiperOrigin-RevId: 569043364

References

#3374 - Make fp8 ops use explicit broadcasting.

Author

a-googler

a-googler

Parents

Loading