burn/crates/burn-candle
Dilshod Tadjibaev cd848b1c94
Add is_nan and contains_nan tensor ops (#2088)
* Add is_nan and contains_nan tensor ops

* Enable nan test for burn-candle

* Disabling tests due to #2089
2024-08-06 12:16:12 -05:00
..
src Add is_nan and contains_nan tensor ops (#2088) 2024-08-06 12:16:12 -05:00
Cargo.toml Fix `DataSerialize` conversion for elements of the same type (#1832) 2024-05-28 18:12:44 -04:00
LICENSE-APACHE Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
LICENSE-MIT Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
README.md [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00

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)