Crate xnn 

A lightweight ML framework built from scratch in Rust with GPU-first architecture.

This library provides GPU-accelerated tensor operations via wgpu, with a focus on simplicity, safety, and cross-platform compatibility.

Types

• Context — GPU context for buffer and pipeline management.
• Buffer — Typed GPU buffer for element data.
• Element — Trait for GPU-compatible types (f32, i32, u32, bool).
• Error — Error type for GPU operations.
• Tensor — N-dimensional array with GPU-accelerated operations.

Re-exports

pub use element::Element;

pub use error::Error;

Modules

element

Traits for GPU-compatible element types.

error

Error types for GPU operations.

Structs

Buffer

Typed GPU buffer for element storage.

Context

GPU device context for buffer and pipeline management.

Tensor

N-dimensional tensor with GPU-backed storage.