[package]
edition = "2021"
rust-version = "1.70"
name = "oxirs-embed"
version = "0.2.1"
authors = ["KitaSan <info@kitasan.io>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Knowledge graph embeddings with TransE, ComplEx, and custom models"
homepage = "https://github.com/cool-japan/oxirs"
readme = "README.md"
keywords = [
"embeddings",
"knowledge-graph",
"machine-learning",
"ai",
"rdf",
]
categories = [
"science",
"algorithms",
"data-structures",
]
license = "Apache-2.0"
repository = "https://github.com/cool-japan/oxirs"
[package.metadata.docs.rs]
all-features = true
rustdoc-args = [
"--cfg",
"docsrs",
]
[features]
advanced-models = [
"rotate",
"conve",
"tucker",
"quatd",
]
api-server = [
"dep:axum",
"dep:tower",
"dep:tower-http",
]
basic-models = [
"transe",
"complex",
"distmult",
"hole",
]
complex = []
conve = ["neural-models"]
cuda = ["scirs2-linalg/cuda"]
custom-models = []
default = ["basic-models"]
distmult = []
gpu = [
"scirs2-core/gpu",
"scirs2-linalg/gpu",
"scirs2-neural/gpu",
]
gpu-acceleration = [
"gpu",
"cuda",
]
graphql = []
hole = []
metal = ["scirs2-linalg/metal"]
multi-modal = []
neural-models = []
ontology-aware = []
opencl = ["scirs2-linalg/opencl"]
parallel = ["scirs2-linalg/parallel"]
quatd = []
rocm = ["scirs2-linalg/rocm"]
rotate = []
simd = ["scirs2-linalg/simd"]
transe = []
tucker = []
[lib]
name = "oxirs_embed"
path = "src/lib.rs"
[[example]]
name = "advanced_embedding_server"
path = "examples/advanced_embedding_server.rs"
[[example]]
name = "advanced_visualization_demo"
path = "examples/advanced_visualization_demo.rs"
[[example]]
name = "basic_usage"
path = "examples/basic_usage.rs"
[[example]]
name = "biomedical_embedding_demo"
path = "examples/biomedical_embedding_demo.rs"
[[example]]
name = "clustering_demo"
path = "examples/clustering_demo.rs"
[[example]]
name = "comprehensive_benchmark"
path = "examples/comprehensive_benchmark.rs"
[[example]]
name = "conve_model_demo"
path = "examples/conve_model_demo.rs"
required-features = ["conve"]
[[example]]
name = "distributed_training_demo"
path = "examples/distributed_training_demo.rs"
required-features = ["basic-models"]
[[example]]
name = "embed_advanced_features_demo"
path = "examples/embed_advanced_features_demo.rs"
[[example]]
name = "embed_basic_usage"
path = "examples/embed_basic_usage.rs"
[[example]]
name = "fine_tuning_demo"
path = "examples/fine_tuning_demo.rs"
[[example]]
name = "gpu_acceleration_demo"
path = "examples/gpu_acceleration_demo.rs"
required-features = ["gpu"]
[[example]]
name = "hole_model_demo"
path = "examples/hole_model_demo.rs"
[[example]]
name = "integrated_ai_platform_demo"
path = "examples/integrated_ai_platform_demo.rs"
required-features = ["gpu"]
[[example]]
name = "interpretability_demo"
path = "examples/interpretability_demo.rs"
[[example]]
name = "link_prediction_demo"
path = "examples/link_prediction_demo.rs"
[[example]]
name = "model_persistence_demo"
path = "examples/model_persistence_demo.rs"
[[example]]
name = "model_selection_demo"
path = "examples/model_selection_demo.rs"
[[example]]
name = "multimodal_embedding_demo"
path = "examples/multimodal_embedding_demo.rs"
[[example]]
name = "performance_profiling_demo"
path = "examples/performance_profiling_demo.rs"
[[example]]
name = "quick_start_guide"
path = "examples/quick_start_guide.rs"
[[example]]
name = "sparql_extension_demo"
path = "examples/sparql_extension_demo.rs"
[[example]]
name = "storage_backend_demo"
path = "examples/storage_backend_demo.rs"
required-features = ["basic-models"]
[[example]]
name = "temporal_embeddings_demo"
path = "examples/temporal_embeddings_demo.rs"
[[example]]
name = "vector_search_demo"
path = "examples/vector_search_demo.rs"
[[test]]
name = "integration_new_features"
path = "tests/integration_new_features.rs"
[[test]]
name = "integration_tests"
path = "tests/integration_tests.rs"
[[bench]]
name = "performance_validation"
path = "benches/performance_validation.rs"
[dependencies.anyhow]
version = "1.0"
[dependencies.async-graphql]
version = "7.0"
[dependencies.async-trait]
version = "0.1"
[dependencies.axum]
version = "0.8"
optional = true
[dependencies.blake3]
version = "1.8"
[dependencies.chrono]
version = "0.4.44"
features = ["serde"]
[dependencies.flate2]
version = "1.1"
[dependencies.futures-util]
version = "0.3"
[dependencies.nalgebra]
version = "0.34"
[dependencies.num_cpus]
version = "1.17"
[dependencies.oxiarc-lz4]
version = "0.2.3"
[dependencies.oxiarc-zstd]
version = "0.2.3"
[dependencies.oxicode]
version = "0.1.1"
features = [
"serde",
"derive",
]
[dependencies.oxirs-vec]
version = "0.2.1"
[dependencies.rayon]
version = "1.11"
[dependencies.regex]
version = "1.12.3"
[dependencies.scirs2-core]
version = "0.3.1"
features = [
"random",
"random",
"array",
"parallel",
"simd",
"profiling",
]
[dependencies.scirs2-linalg]
version = "0.3.1"
[dependencies.scirs2-metrics]
version = "0.3.1"
[dependencies.scirs2-neural]
version = "0.3.1"
[dependencies.scirs2-stats]
version = "0.3.1"
[dependencies.serde]
version = "1.0.228"
features = ["derive"]
[dependencies.serde_json]
version = "1.0.149"
[dependencies.sha2]
version = "0.10.9"
features = ["oid"]
[dependencies.tempfile]
version = "3.27"
[dependencies.thiserror]
version = "2.0"
[dependencies.tokio]
version = "1.50.0"
features = ["full"]
[dependencies.tokio-stream]
version = "0.1"
features = ["sync"]
[dependencies.tower]
version = "0.5"
optional = true
[dependencies.tower-http]
version = "0.6"
features = [
"cors",
"timeout",
"trace",
]
optional = true
[dependencies.tracing]
version = "0.1"
[dependencies.uuid]
version = "1.22"
features = [
"v4",
"serde",
]
[dev-dependencies.criterion]
version = "0.8"
features = ["html_reports"]
[dev-dependencies.tempfile]
version = "3.27"
[dev-dependencies.tracing-subscriber]
version = "0.3"
features = [
"env-filter",
"json",
"fmt",
]