burn-flex

A fast, portable CPU backend for Burn.

Features

• Pure Rust (no C dependencies)
• f16/bf16 support
• SIMD acceleration (NEON on ARM, future AVX on x86)
• 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());