FlashKAN: B-Spline KANs via Truncated Power Form

FlashKAN introduces a new implementation of Kolmogorov‑Arnold Networks (KANs) that replaces the traditional Cox‑de Boor recursion with a truncated power form, allowing each uniform cubic B‑spline to be expressed as five shifted −(x)−^3 terms. The torch.compile‑fused implementation collapses these operations into a single GPU kernel, eliminating recursion, span lookup, and scatter‑gather steps. Additionally, the method includes a bounded‑coordinate stabilization to clamp inputs to [0, k+1], preventing catastrophic cancellation, and provides a production‑ready, open‑source package (pip install flashkan) as a drop‑in replacement for existing KAN layers.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.