[Feature Request / Question] Plotting KAN layer components separately (similar to TruKAN)

I am looking for a way to visualize the individual contributions of a KAN layer's components (SiLU base function and B-spline terms) separately, similar to the approach used in [TruKAN](https://github.com/alibayeh/trukan).