[package]
name = "riemann-flow-gnn"
version = "0.1.0"
edition = "2021"

[dependencies]
plotters = { version = "0.3", default-features = false, features = ["bitmap_backend", "bitmap_encoder"] }
image = "0.24"
serde = { version = "1", features = ["derive"] }
serde_json = "1"
thiserror = "1"
burn = { version = "0.14", features = ["autodiff", "cuda-jit"] }
safetensors = { version = "0.4" }

[dev-dependencies]
approx = "0.5"