deepmd 0.1.0

DeePMD-kit deep potential models as RLX IR graph builders
Documentation 
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

[package]
name = "deepmd"
version.workspace = true
edition.workspace = true
rust-version.workspace = true
license.workspace = true
authors.workspace = true
description = "DeePMD-kit deep potential models as RLX IR graph builders"
repository.workspace = true
homepage.workspace = true
documentation.workspace = true
readme.workspace = true
keywords.workspace = true
categories.workspace = true

[features]
default = []
# Toggle non-CPU backends at build time:
#   cargo build -p deepmd --features metal     # Apple Metal
#   cargo build -p deepmd --features mlx       # Apple MLX
#   cargo build -p deepmd --features cuda      # NVIDIA CUDA (Linux/Windows)
metal = ["rlx-runtime/metal"]
mlx = ["rlx-runtime/mlx"]
cuda = ["rlx-runtime/cuda"]
wgpu = ["rlx-runtime/gpu"]
ane = ["rlx-runtime/ane"]

[dependencies]
rlx-ir   = { workspace = true }
rlx-flow = { workspace = true }
rlx-opt  = { workspace = true }
rlx-runtime = { workspace = true }
rlx-cpu     = { workspace = true }

anyhow   = { workspace = true }
serde    = { workspace = true }
serde_json = { workspace = true }

[dev-dependencies]
rlx-driver = { workspace = true }
criterion = { version = "0.5", default-features = false, features = ["cargo_bench_support"] }

[[bench]]
name = "dp_energy_bench"
harness = false

[lints]
workspace = true