Crate burn_flex 

burn-flex

A fast, portable CPU backend for Burn.

Features

• Pure Rust (no C dependencies)
• f16/bf16 support
• SIMD acceleration via macerator (NEON, AVX2/AVX-512/SSE, SIMD128, scalar fallback)
• Zero-copy tensor views
• Thread-safe by design

Usage

use burn_flex::Flex;
use burn::tensor::Tensor;

let tensor: Tensor<Flex, 2> = Tensor::from_data([[1.0, 2.0], [3.0, 4.0]], &Default::default());

Structs

Flex

The Flex backend, a fast, portable CPU backend for Burn.

FlexDevice

CPU device for the Flex backend.

FlexQTensor

Quantized tensor for the Flex backend.

FlexTensor

CPU tensor primitive for the Flex backend.

Layout

Layout describes how to interpret a linear buffer as an N-dimensional tensor.