[dependencies.candle-core]
optional = true
version = "0.4.1"
[dependencies.candle-flash-attn]
optional = true
version = "0.4.1"
[dependencies.colored]
version = "2.1.0"
[dependencies.dyn-clone]
version = "1.0.16"
[dependencies.half]
features = ["num-traits", "use-intrinsics", "rand_distr"]
version = "2.3.1"
[dependencies.num-traits]
version = "0.2.17"
[dependencies.once_cell]
version = "1.19.0"
[dependencies.paste]
version = "1.0"
[dependencies.safetensors]
version = "0.4.1"
[dependencies.thiserror]
version = "1.0.51"
[dependencies.tracing]
version = "0.1.40"
[features]
accelerate = ["candle-core?/accelerate"]
candle-backend = ["candle-core"]
candle-backend-fa = ["candle-core", "cuda", "candle-flash-attn"]
cuda = ["candle-core?/cuda"]
cudnn = ["candle-core?/cudnn"]
debug-location = []
default = ["candle-backend", "debug-location"]
metal = ["candle-core?/metal"]
mkl = ["candle-core?/mkl"]
[package]
authors = ["cksac <cs.cksac@gmail.com>"]
categories = ["science"]
description = "ML framework with Ergonomic APIs in Rust"
edition = "2021"
homepage = "https://github.com/cksac/rai"
keywords = ["deep-learning", "machine-learning", "tensor"]
license = "MIT OR Apache-2.0"
name = "rai-core"
readme = "README.md"
repository = "https://github.com/cksac/rai"
version = "0.11.0"