Crate briny_ai 

briny_ai

This crate provides a fast, minimal, and modular deep learning backend built in Rust. It features basic tensor operations, automatic differentiation, and support for CPU acceleration via Rayon and AVX2 SIMD. GPU support via WGPU is optional and designed for portability across Intel, AMD, and NVIDIA hardware.

Features

• Tensors: N-dimensional arrays with shape tracking and gradient support
• Autograd: Functional-style forward and backward passes
• Operators: Efficient implementations of matrix multiplication, ReLU, mean squared error, and stochastic gradient descent
• GPU Acceleration: Optional wgpu-powered compute shaders for matrix ops

Modules

backend

Backend selection module.

manual

Tedious manual tensor operations.