burn/examples/custom-wgpu-kernel
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
..
examples Refactor tensor data (#1916) 2024-06-26 20:22:19 -04:00
src Add static tensor quantization (#1963) 2024-07-08 10:16:58 -04:00
Cargo.toml Refactor: split JitKernel and SourceKernel (#1569) 2024-04-05 12:58:10 -04:00