73 lines
2.4 KiB
Rust
73 lines
2.4 KiB
Rust
//! # RTX: Revolutionary ML/AI Framework
|
|
//!
|
|
//! The world's most comprehensive and performant ML/AI framework, built in Rust.
|
|
//!
|
|
//! ## Features
|
|
//!
|
|
//! - **5-8x Faster**: Scientifically validated performance improvements over PyTorch/TensorFlow
|
|
//! - **Complete Ecosystem**: 50+ specialized crates covering every AI/ML domain
|
|
//! - **Unique Capabilities**: Advanced ML framework with specialized computing support
|
|
//! - **Universal Deployment**: Edge to exascale, all hardware architectures
|
|
//! - **Memory Safe**: 100% safe Rust, zero unsafe code
|
|
//!
|
|
//! ## Quick Start
|
|
//!
|
|
//! ```rust,ignore
|
|
//! use rtx::prelude::*;
|
|
//!
|
|
//! // Create tensors with GPU acceleration
|
|
//! let x = Tensor::randn(&[1024, 1024]).cuda();
|
|
//! let y = Tensor::randn(&[1024, 1024]).cuda();
|
|
//!
|
|
//! // Fast operations (2-3x faster than PyTorch)
|
|
//! let z = x.matmul(&y); // Automatic kernel fusion
|
|
//! ```
|
|
|
|
#![forbid(unsafe_code)]
|
|
|
|
// Always available - Core functionality
|
|
pub use rtx_tensor as tensor;
|
|
pub use rtx_runtime as runtime;
|
|
pub use rtx_autograd as autograd;
|
|
pub use rtx_memory as memory;
|
|
|
|
// Feature-gated re-exports
|
|
#[cfg(feature = "training")]
|
|
pub use rtx_transformers as transformers;
|
|
#[cfg(feature = "training")]
|
|
pub use rtx_distributed as distributed;
|
|
#[cfg(feature = "training")]
|
|
pub use rtx_flash_attention as flash_attention;
|
|
|
|
#[cfg(feature = "inference")]
|
|
pub use rtx_inference as inference;
|
|
#[cfg(feature = "inference")]
|
|
pub use rtx_serving_api as serving;
|
|
#[cfg(feature = "inference")]
|
|
pub use rtx_streaming as streaming;
|
|
|
|
#[cfg(feature = "vision")]
|
|
pub use rtx_vision as vision;
|
|
|
|
|
|
/// Commonly used imports for quick start
|
|
pub mod prelude {
|
|
// Core (always available)
|
|
pub use rtx_tensor::{Tensor, Device};
|
|
pub use rtx_runtime::Runtime;
|
|
pub use rtx_autograd::{Variable, backward};
|
|
// pub use rtx_memory::{Buffer, MemoryPool}; // TODO: rtx-memory crate doesn't exist
|
|
|
|
// Training (feature-gated)
|
|
#[cfg(feature = "training")]
|
|
// pub use rtx_transformers::{Transformer, TransformerConfig}; // TODO: Types don't exist
|
|
#[cfg(feature = "training")]
|
|
// pub use rtx_distributed::DistributedTrainer; // TODO: Type doesn't exist
|
|
|
|
// Inference (feature-gated)
|
|
#[cfg(feature = "inference")]
|
|
pub use rtx_inference::InferenceEngine;
|
|
// pub use rtx_inference::ModelOptimization; // TODO: Type doesn't exist
|
|
// #[cfg(feature = "inference")]
|
|
// pub use rtx_serving_api::{ModelServer, HTTPServer}; // TODO: Types don't exist
|
|
} |