burn/burn-core/Cargo.toml

44 lines
1.2 KiB
TOML
Raw Normal View History

2023-01-02 08:21:08 +08:00
[package]
name = "burn-core"
2023-02-13 04:32:29 +08:00
version = "0.5.0"
2023-01-02 08:21:08 +08:00
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
description = "BURN: Burn Unstoppable Rusty Neurons"
repository = "https://github.com/burn-rs/burn-core"
readme = "README.md"
keywords = ["deep-learning", "machine-learning", "tensor", "pytorch", "ndarray"]
categories = ["science"]
license = "MIT/Apache-2.0"
edition = "2021"
[features]
default = []
experimental-named-tensor = ["burn-tensor/experimental-named-tensor"]
[dependencies]
2023-02-13 04:32:29 +08:00
burn-tensor = { version = "0.5.0", path = "../burn-tensor" }
burn-autodiff = { version = "0.5.0", path = "../burn-autodiff" }
burn-dataset = { version = "0.5.0", path = "../burn-dataset", default-features = false }
burn-derive = { version = "0.5.0", path = "../burn-derive" }
2023-01-02 08:21:08 +08:00
# Utilities
derive-new = "0.5.9"
rand = "0.8.5"
log = "0.4.17"
# Serialize Deserialize
serde = { version = "1.0.151", features = ["derive"] }
serde_json = "1.0.91"
flate2 = "1.0.25"
# Parameter & Optimization
nanoid = "0.4.0"
[dev-dependencies]
2023-02-13 04:32:29 +08:00
burn-dataset = { version = "0.5.0", path = "../burn-dataset", features = [
2023-01-02 08:21:08 +08:00
"fake",
] }
burn-ndarray = { version = "0.5.0", path = "../burn-ndarray", features = [
"std"
] }