burn/burn-book
Guillaume Lagrange c0211e2f94
Add static tensor quantization (#1963)
* Add QuantizationBackend, QTensorOps and QTensor

* Refactor QTensorOps as part of Backend trait

* Add tensor dequantize, QFloat dtype and default affine/symmetric quant

* Add ndarray default quantization implementation

* Fix clippy

* Add rayon parallel iter

* Add quantization operations to book

* Add q_shape and q_device ops to avoid converting the tensor just to get attributes

* Implement autodiff grad ops

* Mark autodiff todo for QAT

* Remove note

* Add q_inner and q_from_inner
2024-07-08 10:16:58 -04:00
..
src Add static tensor quantization (#1963) 2024-07-08 10:16:58 -04:00
.gitignore The Burn Book (#625) 2023-08-10 14:17:23 -04:00
.prettierrc.json Doc/book/getting started (#744) 2023-08-31 14:08:13 -04:00
LICENSE-APACHE License fixes (#648) 2023-08-16 12:45:35 -04:00
LICENSE-MIT License fixes (#648) 2023-08-16 12:45:35 -04:00
book.toml [guide] Remove ambiguity lib vs. executable (#1649) 2024-04-26 15:42:02 -04:00