-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
26 lines (22 loc) · 723 Bytes
/
Cargo.toml
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
26
[package]
name = "candle-birnn"
version = "0.2.3"
edition = "2021"
description = "implement Pytorch LSTM and BiDirectional LSTM with Candle"
readme = "README.md"
repository = "https://github.com/kigichang/candle-birnn"
license-file = "LICENSE"
keywords = ["candle", "lstm", "pytorch"]
[dependencies]
candle-core = "0.7.0"
candle-nn = "0.7.0"
candle-metal-kernels = {version = "0.7.0", optional = true}
candle-kernels = {version = "0.7.0", optional = true}
[dev-dependencies]
anyhow = "1.0.86"
[features]
default = []
accelerate = ["candle-core/accelerate", "candle-nn/accelerate"]
mkl = ["candle-core/mkl", "candle-nn/mkl"]
metal = ["candle-core/metal", "candle-nn/metal"]
cuda = ["candle-core/cuda", "candle-nn/cuda"]