Skip to main content

Module inference

Module inference 

Source
Expand description

Pure Rust inference for WASM compilation Pure Rust inference module for WASM compatibility

This module provides a lightweight neural network inference engine that can be compiled to WebAssembly. It deliberately avoids the Burn tensor stack used on the training side (see src/train/, src/policy/) so the WASM bundle stays small — Burn’s WebGPU backend has a non-trivial WASM bundle-size cost on a pre-1.0 backend. See docs/WASM_ROADMAP.md and Burn migration epic #65 (phase 6 closure) for the rationale.

Forward pass only — no autodiff, no training.

Modules§

nn
Neural network utilities for pure Rust inference
snake
Snake CNN inference for WASM
weights
Weight export/import utilities for trained models

Structs§

ExportedModel
Exported model format for WASM inference
LayerWeights
Weights and biases for a single layer