dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/burn-candle
History
Joshua Ferguson 4a70a0f8bc
renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 		2024-01-27 10:04:50 -05:00
..
src renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 2024-01-27 10:04:50 -05:00
Cargo.toml Add support for loading PyTorch `.pt` (weights/states) files directly to model's record (#1085) 2024-01-25 10:20:09 -05:00
README.md Chore/release (#1031) 2023-12-01 14:33:28 -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)