Skip to content

Commit ea191d9

Browse files
author
“thucydides”
committed
Create RusTorch crate: a facade library merging core, nn, vision, text, and wasm
1 parent 331a543 commit ea191d9

2 files changed

Lines changed: 47 additions & 0 deletions

File tree

rus-torch/Cargo.toml

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
[package]
2+
name = "rus-torch"
3+
version = "0.1.0"
4+
edition = "2021"
5+
authors = ["Trae AI <trae@example.com>"]
6+
description = "A comprehensive deep learning framework in Rust, merging core, nn, vision, text, and wasm"
7+
license = "MIT OR Apache-2.0"
8+
repository = "https://github.com/Genius-apple/RusTorch"
9+
keywords = ["deep-learning", "machine-learning", "neural-network", "computer-vision", "nlp"]
10+
categories = ["science", "computer-vision", "text-processing", "wasm"]
11+
12+
[dependencies]
13+
rustorch-core = { path = "../rustorch-core", version = "0.1.0" }
14+
rustorch-nn = { path = "../rustorch-nn", version = "0.1.0" }
15+
rustorch-vision = { path = "../rustorch-vision", version = "0.1.0", optional = true }
16+
rustorch-text = { path = "../rustorch-text", version = "0.1.0", optional = true }
17+
rustorch-wasm = { path = "../rustorch-wasm", version = "0.1.0", optional = true }
18+
19+
[features]
20+
default = ["vision", "text"]
21+
vision = ["dep:rustorch-vision"]
22+
text = ["dep:rustorch-text"]
23+
wasm = ["dep:rustorch-wasm"]

rus-torch/src/lib.rs

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
//! RusTorch: A comprehensive deep learning framework in Rust.
2+
//!
3+
//! This crate re-exports the core components:
4+
//! - `rustorch_core` as `core`
5+
//! - `rustorch_nn` as `nn`
6+
//! - `rustorch_vision` as `vision` (feature "vision")
7+
//! - `rustorch_text` as `text` (feature "text")
8+
//! - `rustorch_wasm` as `wasm` (feature "wasm")
9+
10+
pub use rustorch_core as core;
11+
pub use rustorch_nn as nn;
12+
13+
#[cfg(feature = "vision")]
14+
pub use rustorch_vision as vision;
15+
16+
#[cfg(feature = "text")]
17+
pub use rustorch_text as text;
18+
19+
#[cfg(feature = "wasm")]
20+
pub use rustorch_wasm as wasm;
21+
22+
// Convenience re-exports
23+
pub use core::Tensor;
24+
pub use nn::Module;

0 commit comments

Comments
 (0)