mirror of https://github.com/tracel-ai/burn.git
c0211e2f94
* 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 |
||
---|---|---|
.. | ||
src | ||
Cargo.toml | ||
LICENSE-APACHE | ||
LICENSE-MIT | ||
README.md |
README.md
Burn Candle Backend
This crate provides a backend for Burn based on the Candle framework.
It is still in alpha stage, not all operations are supported. It is usable for some use cases, like for inference.
It can be used with CPU or CUDA. On macOS computations can be accelerated by using the Accelerate framework.
Feature Flags
The following features are supported:
cuda
- Cuda GPU device (NVIDIA only)accelerate
- Accelerate framework (macOS only)