Architecture diagram

NNx is organized around two public entry points (NNModel / Trainer), a single extensibility hook (train_step_fn / trainer_step_fn), and content-addressed persistence under runs/<id>/. The diagram below shows the eight-layer top-to-bottom flow; see Concepts §1 for the full written breakdown.

Standalone version

Open the standalone diagram page in a new tab — it renders the same flow with a per-layer explanation card grid.