Crate xnn 

A lightweight ML framework built with Rust.

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

Types

• GpuContext — GPU context for buffer and pipeline management.
• Buffer — typed GPU buffer for element data.
• Element — trait for GPU-compatible types (f32, i32, u32).
• FloatElement — marker trait for floating-point types (f32).
• Error — error type for GPU operations.

Feature flags

• unstable-kernels — exposes the kernel module with GPU compute kernels.

Modules

kernel

GPU compute kernels.

Structs

Buffer

Typed wrapper around a GPU buffer.

GpuContext

Central GPU context for buffer and pipeline management.

Enums

Error

Errors that can occur during GPU operations.

Traits

Element

Trait for GPU-compatible element types.

FloatElement

Trait for floating-point GPU-compatible types.