23 lines
694 B
Rust
23 lines
694 B
Rust
//! RTX Core - Essential tensor operations and runtime
|
|
//!
|
|
//! This crate provides minimal RTX functionality for users who only need:
|
|
//! - High-performance tensor operations (2-3x faster than PyTorch)
|
|
//! - GPU runtime with CUDA/ROCm support
|
|
//! - Automatic differentiation
|
|
//! - Advanced memory management
|
|
|
|
#![forbid(unsafe_code)]
|
|
|
|
pub use rtx_autograd as autograd;
|
|
pub use rtx_memory as memory;
|
|
pub use rtx_runtime as runtime;
|
|
pub use rtx_tensor as tensor;
|
|
|
|
/// Essential imports for tensor computing
|
|
pub mod prelude {
|
|
pub use rtx_autograd::backward;
|
|
pub use rtx_memory::{MemoryBlock, MemoryPoolManager};
|
|
pub use rtx_runtime::Runtime;
|
|
pub use rtx_tensor::{Device, Tensor};
|
|
}
|