Awni Hannun 70560b6bd5 Add mode parameter for quantization (#2499) ...

* add mode parameter for quantization

* mxfp4 quantize/dequantize + start of optional biases

* mxfp4 works

* speedup

* cpu mxfp4

* fix

* fix test tol

* fix

* refactor

* add quant mode enum

2025-08-28 06:45:26 -07:00

..

layers

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -07:00

__init__.py

Ring docs (#1829)

2025-02-28 11:34:21 -08:00

init.py

catch stream errors earlier to avoid aborts (#1801)

2025-01-27 14:05:43 -08:00

losses.py

update the formula of smooth_l1_loss (#1986)

2025-03-21 06:25:23 -07:00

utils.py

Remove stream from average grads so it uses default (#2532)

2025-08-25 15:56:29 -07:00