dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/burn-ndarray
History
Will Brickner 03af140e12
Implement Quiet Softmax (`Attention Is Off By One`) (#692) 
* Added quiet_softmax

* Undid bad formatting

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 		2023-11-30 12:58:30 -05:00
..
src Implement Quiet Softmax (`Attention Is Off By One`) (#692) 2023-11-30 12:58:30 -05:00
Cargo.toml Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -04:00
LICENSE-APACHE Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
LICENSE-MIT Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
README.md Update readme (#962) 2023-11-17 13:04:41 -05:00
build.rs Add accelerate framework support for burn-ndarray (#183) (#184) 2023-02-28 09:09:47 -05:00

README.md

Burn NdArray

Burn ndarray backend

Current Crates.io Version license

Feature Flags

This crate can be used without the standard library (#![no_std]) with alloc by disabling the default std feature.

The following flags support various BLAS options:

  • blas-accelerate - Accelerate framework (macOS only)
  • blas-netlib - Netlib
  • blas-openblas - OpenBLAS static linked
  • blas-openblas-system - OpenBLAS from the system

Note, under the no_std mode, a random seed is generated during the build time if the seed is not initialized by by Backend::seed method.

Platform Support

Option CPU GPU Linux MacOS Windows Android iOS WASM
Pure Rust Yes No Yes Yes Yes Yes Yes Yes
Accelerate Yes No No Yes No No Yes No
Netlib Yes No Yes Yes Yes No No No
Openblas Yes No Yes Yes Yes Yes Yes No