Nadaraya Watson

1 long-form post on Nadaraya Watson: machine-learning research by Taha Bouhsine, each built around live, in-browser interactive visualizations.

1 post tagged #nadaraya-watson.

Jun 4, 2026

The Prototype Readout in JAX/Flax NNX

A runnable companion to The Readout is a Convex Combination of Prototypes: read the columns of W_out as output prototypes in Flax NNX, measure the convex/conic/affine/linear regimes numerically, then build a Nadaraya–Watson kernel readout that is convex by construction (nonnegative weights that sum to one, a point that never leaves the prototype hull), with the nonnegativity-vs-positive-definiteness distinction checked in code.