[package]
edition = "2024"
name = "fuel-transformers"
version = "0.20.0"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Minimalist ML framework."
readme = "README.md"
keywords = [
"blas",
"tensor",
"machine-learning",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/ciresnave/fuel"
resolver = "2"
[features]
accelerate = [
"dep:accelerate-src",
"fuel-tensor/accelerate",
]
cuda = ["fuel-tensor/cuda"]
cudnn = ["fuel-tensor/cudnn"]
default = []
metal = ["fuel-tensor/metal"]
mkl = [
"dep:intel-mkl-src",
"fuel-tensor/mkl",
]
[lib]
name = "fuel_transformers"
path = "src/lib.rs"
[[test]]
name = "config_parser_membership"
path = "tests/config_parser_membership.rs"
[[test]]
name = "gap314_spatial_population"
path = "tests/gap314_spatial_population.rs"
[[test]]
name = "generation_tests"
path = "tests/generation_tests.rs"
[[test]]
name = "nms_tests"
path = "tests/nms_tests.rs"
[[test]]
name = "phase6a_anchor_oracle"
path = "tests/phase6a_anchor_oracle.rs"
[[test]]
name = "phase6b_cuda_anchor"
path = "tests/phase6b_cuda_anchor.rs"
[[test]]
name = "qwen3_cpu_yardstick"
path = "tests/qwen3_cpu_yardstick.rs"
[dependencies.accelerate-src]
version = "0.3.2"
optional = true
[dependencies.byteorder]
version = "1.4.3"
[dependencies.fancy-regex]
version = "0.19.0"
[dependencies.fuel-cpu-backend]
version = "0.20.0"
[dependencies.fuel-dispatch]
version = "0.20.0"
[dependencies.fuel-graph]
version = "0.20.0"
[dependencies.fuel-ir]
version = "0.20.0"
[dependencies.fuel-loaders]
version = "0.20.0"
[dependencies.fuel-model-llama]
version = "0.20.0"
[dependencies.fuel-model-phi]
version = "0.20.0"
[dependencies.fuel-quantized]
version = "0.20.0"
[dependencies.fuel-tensor]
version = "0.20.0"
[dependencies.half]
version = "2.5.0"
features = [
"num-traits",
"use-intrinsics",
"rand_distr",
"bytemuck",
]
[dependencies.hf-hub]
version = "0.5.0"
[dependencies.image]
version = "0.25.2"
features = [
"jpeg",
"png",
]
default-features = false
[dependencies.intel-mkl-src]
version = "0.8.1"
features = ["mkl-static-lp64-iomp"]
optional = true
[dependencies.num-traits]
version = "0.2.15"
[dependencies.rand]
version = "0.10"
[dependencies.rand_distr]
version = "0.6"
[dependencies.rayon]
version = "1.12"
[dependencies.safetensors]
version = "0.8.0"
[dependencies.serde]
version = "1.0.171"
features = ["derive"]
[dependencies.serde_json]
version = "1.0.99"
[dependencies.serde_plain]
version = "1.0.2"
[dependencies.tokenizers]
version = "0.23.1"
default-features = false
[dependencies.tracing]
version = "0.1.37"
[dev-dependencies.anyhow]
version = "1"
features = ["backtrace"]