5 Commits

Author	SHA1	Message	Date
VED	1eaf4d7418	add: support mxfp4 axo (#3375) ... * mxfp4 axo * import lint * test for qat mxfp4 * config for mxfp4 * add qat: * pass base config * MXFakeQuantizeConfig * lint * tune config so it fits in 32GB VRAM --------- Co-authored-by: Wing Lian <wing@axolotl.ai>	2026-03-05 13:40:45 -05:00
VED	a6bfbe3400	torch_dtype -> dtype (#3177) ... * torch_dtype -> dtype * torch_dtype -> dtype	2025-10-01 15:02:51 +07:00
salman	58d67bf98d	Migrate QAT API; fix axolotl quantize for QAT-ed models; add NVFP4 (#3107)	2025-09-12 10:55:50 +01:00
Dan Saunders	79ddaebe9a	Add ruff, remove black, isort, flake8, pylint (#3092) ... * black, isort, flake8 -> ruff * remove unused * add back needed import * fix	2025-08-23 23:37:33 -04:00
salman	5fca214108	QAT (#2590) ... QAT and quantization w/torchao	2025-05-28 12:35:47 +01:00