Implementing Kolmogorov-Arnold Networks as Dense GEMMs in JAX: A Neural QMC Experiment

Update (Sep 2026): After publishing, I found problems in the design described here, and I have revised this post. The ε-manifold mapping (originally Sections 1–2) was unnecessary. Real solid harmonics