mirror of https://github.com/tracel-ai/burn.git
0d5025edbb
* Move QuantizationScheme to burn-tensor * Refactor QuantizedTensorPrimitive to include the quantization strategy * Fix QFloat tensor data display * Refactor quantization methods to use scheme and qparams (on backend device) * Fix clippy * Fix fmt * Add qtensor primitive tests |
||
---|---|---|
.. | ||
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)