12 Commits

Author	SHA1	Message	Date
shuw	1fe24ca755	Improve based on review 1	2025-03-20 23:26:21 +00:00
shuw	549c669451	Straight-through estimator for nvfp4	2025-03-20 19:30:14 +00:00
jax authors	726f49cbca	Merge pull request #26944 from wenscarl:wenscarl/nvfp4 ... PiperOrigin-RevId: 736620378	2025-03-13 13:30:46 -07:00
shuw	f9aef8a189	Support nvfp4	2025-03-11 19:33:25 +00:00
Shu Wang	7f0a5bc83e	Add apache header.	2025-02-26 15:26:56 -06:00
shuw	17088e9025	Improve after review # 2	2025-02-26 04:48:25 +00:00
shuw	681ee18436	Fix CI	2025-02-25 17:15:31 +00:00
shuw	d9456f36b3	refresh CI	2025-02-24 17:34:06 +00:00
Shu Wang	08012e9c01	Conditionally create mxfp8_configs.	2025-02-21 23:08:22 -06:00
shuw	bfb9d3ca4b	Improve based on comment # 1	2025-02-21 17:32:57 +00:00
shuw	332af58765	block_scale_config	2025-02-13 04:35:06 +00:00
shuw	061d4acbfb	Scaled matmul for mxfp8	2025-02-05 23:25:51 +00:00