[package]
edition = "2021"
name = "tsai_models"
version = "0.1.2"
authors = ["tsai-rs contributors"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Model zoo for tsai-rs: CNN, Transformer, ROCKET, RNN, and Tabular architectures"
readme = "README.md"
license = "Apache-2.0"
repository = "https://github.com/TuringWorks/tsai-rs"
[features]
backend-ndarray = ["tsai_core/backend-ndarray"]
backend-tch = ["tsai_core/backend-tch"]
backend-wgpu = ["tsai_core/backend-wgpu"]
default = ["backend-ndarray"]
[lib]
name = "tsai_models"
path = "src/lib.rs"
[dependencies.burn]
version = "0.16"
[dependencies.rand]
version = "0.8"
[dependencies.rand_chacha]
version = "0.3"
[dependencies.schemars]
version = "0.8"
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0"
[dependencies.thiserror]
version = "2.0"
[dependencies.tsai_core]
version = "0.1.2"
[dev-dependencies.approx]
version = "0.5"
[dev-dependencies.burn-ndarray]
version = "0.16"