summaryrefslogtreecommitdiff
path: root/candle-pyo3/Cargo.toml
blob: a98d47c2cd8af3807bca56b5f9910099d291af56 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
[package]
name = "candle-pyo3"
version = "0.1.0"
edition = "2021"

description = "PyO3 bindings for the candle ML framework."
repository = "https://github.com/LaurentMazare/candle"
keywords = ["blas", "tensor", "machine-learning"]
categories = ["science"]
license = "MIT/Apache-2.0"
readme = "README.md"

[lib]
name = "candle"
crate-type = ["cdylib"]
doc = false

[dependencies]
candle = { path = "../candle-core", package = "candle-core" }
pyo3 = { version = "0.19.0", features = ["extension-module"] }
half = { workspace = true }

[features]
default = []
cuda = ["candle/cuda"]