Commit Graph

6 Commits

Author SHA1 Message Date
Chunyang Wen 2aedf3e791 Minor refactor for tree_map and tree_unflatten (#311)
* Minor refact for tree_map and tree_unflatten

* Remove the if statement

---------

Co-authored-by: Chunyang Wen <chunyang_wen@apple.com>
2023-12-28 20:55:10 -08:00
Chunyang Wen 473b6b43b4 Use defaultdict (#307)
Co-authored-by: Chunyang Wen <chunyang_wen@apple.com>
2023-12-28 14:46:13 -08:00
Angelos Katharopoulos 57fe918cf8 Adds C++ and nn quantization utilities (#230)
* Add C++ de-/quantize ops
* Add quantize functions to the docs and tests
* Add a QuantizedLinear module
2023-12-20 14:17:38 -08:00
Awni Hannun db487e6b1a format 2023-11-30 11:50:50 -08:00
Awni Hannun 46a39e5b1f copyright + ack 2023-11-30 11:12:53 -08:00
Angelos Katharopoulos d1f86272a2 angelos's commit files 2023-11-29 10:42:59 -08:00