38 lines
1.1 KiB
TOML
38 lines
1.1 KiB
TOML
[package]
|
|
name = "hanzo-transformers"
|
|
version = "0.11.9"
|
|
edition = "2021"
|
|
description = "Multi-backend tensor & ML framework for Rust — part of the Hanzo ML stack."
|
|
repository = "https://github.com/hanzoai/ml"
|
|
keywords = ["blas", "tensor", "machine-learning"]
|
|
categories = ["science"]
|
|
license = "BSD-3-Clause OR Apache-2.0"
|
|
readme = "README.md"
|
|
|
|
[dependencies]
|
|
accelerate-src = { workspace = true, optional = true }
|
|
byteorder = { workspace = true }
|
|
hanzo-ml = { workspace = true }
|
|
hanzo-nn = { workspace = true }
|
|
fancy-regex = { workspace = true }
|
|
intel-mkl-src = { workspace = true, optional = true }
|
|
num-traits = { workspace = true }
|
|
rand = { workspace = true }
|
|
rayon = { workspace = true }
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
serde_plain = { workspace = true }
|
|
tracing = { workspace = true }
|
|
|
|
[features]
|
|
default = []
|
|
accelerate = ["dep:accelerate-src", "hanzo-ml/accelerate", "hanzo-nn/accelerate"]
|
|
cuda = ["hanzo-ml/cuda", "hanzo-nn/cuda"]
|
|
cudnn = ["hanzo-ml/cudnn", "hanzo-nn/cudnn"]
|
|
flash-attn = ["cuda"]
|
|
mkl = ["dep:intel-mkl-src", "hanzo-ml/mkl", "hanzo-nn/mkl"]
|
|
metal = ["hanzo-ml/metal", "hanzo-nn/metal"]
|
|
|
|
[lints]
|
|
workspace = true
|