Cargo.toml

[package]
name = "burn-efficient-kan"
version = "0.2.0"
edition = "2021"
license = "MIT"
authors = ["Vladislav Grechannik <vgechannik@gmail.com>"]
description = "An efficient pure-Rust implementation of Kolmogorov-Arnold Network (KAN)."
repository = "https://github.com/VlaDexa/burn-efficient-kan"
keywords = [
	"machine-learning",
	"deep-learning",
	"neural-networks",
	"kan",
	"burn",
]
categories = ["science"]

[features]
default = []

netlib = ["ndarray-linalg/netlib"]
openblas = ["ndarray-linalg/openblas"]
intel-mkl = ["ndarray-linalg/intel-mkl"]

netlib-static = ["ndarray-linalg/netlib-static"]
netlib-system = ["ndarray-linalg/netlib-system"]

openblas-static = ["ndarray-linalg/openblas-static"]
openblas-system = ["ndarray-linalg/openblas-system"]

intel-mkl-static = ["ndarray-linalg/intel-mkl-static"]
intel-mkl-system = ["ndarray-linalg/intel-mkl-system"]

[dependencies]
ndarray = "0.15"
ndarray-linalg = "0.16"
burn = "0.14"

[dev-dependencies.burn]
version = "0.14"
features = ["wgpu", "train", "vision", "tui"]

[dev-dependencies.ndarray-linalg]
version = "0.16"
features = ["netlib-system"]

[[example]]
name = "mnist"
path = "examples/mnist/main.rs"