mirror of https://github.com/tracel-ai/burn.git
44 lines
1.4 KiB
TOML
44 lines
1.4 KiB
TOML
[package]
|
|
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
|
|
categories = ["science", "no-std", "embedded", "wasm"]
|
|
description = """
|
|
This library provides multiple tensor implementations hidden behind
|
|
an easy to use API that supports reverse mode automatic differentiation.
|
|
"""
|
|
edition = "2021"
|
|
keywords = ["deep-learning", "machine-learning", "tensor", "pytorch", "ndarray"]
|
|
license = "MIT/Apache-2.0"
|
|
name = "burn-tensor"
|
|
readme = "README.md"
|
|
repository = "https://github.com/burn-rs/burn/tree/main/burn-tensor"
|
|
version = "0.8.0"
|
|
|
|
[features]
|
|
default = ["std"]
|
|
experimental-named-tensor = []
|
|
export_tests = ["burn-tensor-testgen"]
|
|
std = [
|
|
"rand/std",
|
|
"half/std",
|
|
"half/serde", # TODO: set default when https://github.com/starkat99/half-rs/issues/84 is fixed
|
|
]
|
|
|
|
[dependencies]
|
|
burn-tensor-testgen = {path = "../burn-tensor-testgen", version = "0.8.0", optional = true}
|
|
|
|
derive-new = {workspace = true}
|
|
half = {workspace = true}
|
|
libm = {workspace = true}# no_std is supported by default
|
|
num-traits = {workspace = true}
|
|
rand = {workspace = true}
|
|
rand_distr = {workspace = true}# use instead of statrs because it supports no_std
|
|
|
|
# The same implementation of HashMap in std but with no_std support (only needs alloc crate)
|
|
hashbrown = {workspace = true}# no_std compatible
|
|
|
|
# Serialization
|
|
serde = {workspace = true}
|
|
|
|
[dev-dependencies]
|
|
rand = {workspace = true, features = ["std", "std_rng"]}# Default enables std
|