Index

_

__call__() (jaxkan.layers.Chebyshev.ChebyshevLayer method) (jaxkan.layers.Dense.DenseLayer method) (jaxkan.layers.Fourier.FourierLayer method) (jaxkan.layers.Legendre.LegendreLayer method) (jaxkan.layers.RBF.RBFLayer method) (jaxkan.layers.Sine.SineLayer method) (jaxkan.layers.Spline.BaseLayer method) (jaxkan.layers.Spline.SplineLayer method) (jaxkan.models.ActNet.ActLayer method) (jaxkan.models.ActNet.ActNet method) (jaxkan.models.KAN.KAN method) (jaxkan.models.KKAN.ChebyshevEmbedding method) (jaxkan.models.KKAN.InnerBlock method) (jaxkan.models.KKAN.KKAN method) (jaxkan.models.KKAN.OuterBlock method) (jaxkan.models.RGAKAN.RGABlock method) (jaxkan.models.RGAKAN.RGAKAN method) (jaxkan.models.utils.PeriodEmbedder method) (jaxkan.models.utils.RFFEmbedder method) __init__() (jaxkan.grids.BaseGrid.BaseGrid method) (jaxkan.grids.RBFGrid.RBFGrid method) (jaxkan.grids.SplineGrid.SplineGrid method) (jaxkan.layers.Chebyshev.ChebyshevLayer method) (jaxkan.layers.Dense.DenseLayer method) (jaxkan.layers.Fourier.FourierLayer method) (jaxkan.layers.Legendre.LegendreLayer method) (jaxkan.layers.RBF.RBFLayer method) (jaxkan.layers.Sine.SineLayer method) (jaxkan.layers.Spline.BaseLayer method) (jaxkan.layers.Spline.SplineLayer method) (jaxkan.models.ActNet.ActLayer method) (jaxkan.models.ActNet.ActNet method) (jaxkan.models.KAN.KAN method) (jaxkan.models.KKAN.ChebyshevEmbedding method) (jaxkan.models.KKAN.InnerBlock method) (jaxkan.models.KKAN.KKAN method) (jaxkan.models.KKAN.OuterBlock method) (jaxkan.models.RGAKAN.RGABlock method) (jaxkan.models.RGAKAN.RGAKAN method) (jaxkan.models.utils.PeriodEmbedder method) (jaxkan.models.utils.RFFEmbedder method)

A

activation (jaxkan.layers.Dense.DenseLayer attribute) (jaxkan.models.KKAN.InnerBlock attribute) ActLayer (class in jaxkan.models.ActNet) ActNet (class in jaxkan.models.ActNet)

adam_transition() (in module jaxkan.layers.utils) add_bias (jaxkan.models.ActNet.ActNet attribute) alpha (jaxkan.models.RGAKAN.RGABlock attribute) apply_rba_weights() (in module jaxkan.pikan.adaptive) axes (jaxkan.models.utils.PeriodEmbedder attribute)

B

b (jaxkan.layers.Dense.DenseLayer attribute) B (jaxkan.models.utils.RFFEmbedder attribute) BaseGrid (class in jaxkan.grids.BaseGrid) BaseLayer (class in jaxkan.layers.Spline) basis() (jaxkan.layers.Chebyshev.ChebyshevLayer method) (jaxkan.layers.Fourier.FourierLayer method) (jaxkan.layers.Legendre.LegendreLayer method) (jaxkan.layers.RBF.RBFLayer method) (jaxkan.layers.Sine.SineLayer method) (jaxkan.layers.Spline.BaseLayer method) (jaxkan.layers.Spline.SplineLayer method) (jaxkan.models.ActNet.ActLayer method)

batched_frob() (in module jaxkan.models.utils) beta (jaxkan.models.ActNet.ActLayer attribute) (jaxkan.models.RGAKAN.RGABlock attribute) bias (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Fourier.FourierLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) biases (jaxkan.models.ActNet.ActNet attribute) blocks (jaxkan.models.RGAKAN.RGAKAN attribute)

C

C (jaxkan.models.KKAN.ChebyshevEmbedding attribute) c_basis (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) c_cos (jaxkan.layers.Fourier.FourierLayer attribute) c_ext (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute)

c_res (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) c_sin (jaxkan.layers.Fourier.FourierLayer attribute) c_spl (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) ChebyshevEmbedding (class in jaxkan.models.KKAN) ChebyshevLayer (class in jaxkan.layers.Chebyshev) count_params() (in module jaxkan.models.utils)

D

D (jaxkan.grids.RBFGrid.RBFGrid attribute) (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Fourier.FourierLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.models.RGAKAN.RGAKAN attribute)

D_e (jaxkan.models.KKAN.ChebyshevEmbedding attribute) DenseLayer (class in jaxkan.layers.Dense)

F

FE (jaxkan.models.RGAKAN.RGAKAN attribute) flavor (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute)

FourierLayer (class in jaxkan.layers.Fourier)

G

G (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute) g (jaxkan.layers.Dense.DenseLayer attribute) get_ac_res() (in module jaxkan.pikan.pde) get_activation() (in module jaxkan.models.utils) get_adam() (in module jaxkan.models.utils) get_advection_res() (in module jaxkan.pikan.pde) get_burgers_res() (in module jaxkan.pikan.pde) get_causal_weights() (in module jaxkan.pikan.adaptive) get_colloc_indices() (in module jaxkan.pikan.adaptive) get_collocs_grid() (in module jaxkan.pikan.sampling) get_collocs_random() (in module jaxkan.pikan.sampling) get_collocs_sobol() (in module jaxkan.pikan.sampling) get_complexity() (in module jaxkan.models.utils) get_diffusion_res() (in module jaxkan.pikan.pde) get_frob() (in module jaxkan.models.utils) get_helmholtz_res() (in module jaxkan.pikan.pde)

get_kdv_res() (in module jaxkan.pikan.pde) get_ks_res() (in module jaxkan.pikan.pde) get_layer() (in module jaxkan.layers.__init__) get_lbfgs() (in module jaxkan.models.utils) get_poisson_res() (in module jaxkan.pikan.pde) get_rad_indices() (in module jaxkan.pikan.adaptive) get_rad_probabilities() (in module jaxkan.pikan.adaptive) get_sg_res() (in module jaxkan.pikan.pde) get_wave_res() (in module jaxkan.pikan.pde) grid (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) grid_e (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.grids.RBFGrid.RBFGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute) grid_range (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.grids.RBFGrid.RBFGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute)

H

hidden_layers (jaxkan.models.KKAN.InnerBlock attribute)

I

inner_blocks (jaxkan.models.KKAN.KKAN attribute) InnerBlock (class in jaxkan.models.KKAN) input_embedding (jaxkan.models.KKAN.InnerBlock attribute) input_layer (jaxkan.models.KKAN.InnerBlock attribute) input_proj (jaxkan.models.ActNet.ActNet attribute)

InputLayer (jaxkan.models.RGAKAN.RGABlock attribute) interpolate_moments() (in module jaxkan.layers.utils) item (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.grids.RBFGrid.RBFGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute)

J

jaxkan.grids.BaseGrid module jaxkan.grids.RBFGrid module jaxkan.grids.SplineGrid module jaxkan.layers.__init__ module jaxkan.layers.Chebyshev module jaxkan.layers.Dense module jaxkan.layers.Fourier module jaxkan.layers.Legendre module jaxkan.layers.RBF module jaxkan.layers.Sine module jaxkan.layers.Spline module

jaxkan.layers.utils module jaxkan.models.ActNet module jaxkan.models.KAN module jaxkan.models.KKAN module jaxkan.models.RGAKAN module jaxkan.models.utils module jaxkan.pikan.adaptive module jaxkan.pikan.pde module jaxkan.pikan.sampling module jaxkan.pikan.utils module

K

k (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute)

KAN (class in jaxkan.models.KAN) kernel (jaxkan.layers.RBF.RBFLayer attribute) KKAN (class in jaxkan.models.KKAN)

L

Lambda (jaxkan.models.ActNet.ActLayer attribute) layer (jaxkan.models.KKAN.OuterBlock attribute) layers (jaxkan.models.ActNet.ActNet attribute) (jaxkan.models.KAN.KAN attribute)

LegendreLayer (class in jaxkan.layers.Legendre) lr_anneal() (in module jaxkan.pikan.adaptive)

M

model_eval() (in module jaxkan.pikan.utils) module jaxkan.grids.BaseGrid jaxkan.grids.RBFGrid jaxkan.grids.SplineGrid jaxkan.layers.__init__ jaxkan.layers.Chebyshev jaxkan.layers.Dense jaxkan.layers.Fourier jaxkan.layers.Legendre jaxkan.layers.RBF jaxkan.layers.Sine jaxkan.layers.Spline jaxkan.layers.utils jaxkan.models.ActNet jaxkan.models.KAN jaxkan.models.KKAN jaxkan.models.RGAKAN jaxkan.models.utils jaxkan.pikan.adaptive jaxkan.pikan.pde jaxkan.pikan.sampling jaxkan.pikan.utils

N

n_hidden (jaxkan.models.RGAKAN.RGAKAN attribute) n_in (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Fourier.FourierLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) (jaxkan.models.KKAN.KKAN attribute)

n_nodes (jaxkan.grids.RBFGrid.RBFGrid attribute) (jaxkan.grids.SplineGrid.SplineGrid attribute) n_out (jaxkan.grids.BaseGrid.BaseGrid attribute) (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Fourier.FourierLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute)

O

omega (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.models.ActNet.ActLayer attribute) omega0 (jaxkan.models.ActNet.ActNet attribute) OutBasis (jaxkan.models.RGAKAN.RGAKAN attribute) outer_block (jaxkan.models.KKAN.KKAN attribute)

OuterBlock (class in jaxkan.models.KKAN) OutLayer (jaxkan.models.RGAKAN.RGAKAN attribute) output_embedding (jaxkan.models.KKAN.InnerBlock attribute) output_layer (jaxkan.models.KKAN.InnerBlock attribute) output_proj (jaxkan.models.ActNet.ActNet attribute) OutputLayer (jaxkan.models.RGAKAN.RGABlock attribute)

P

PE (jaxkan.models.RGAKAN.RGAKAN attribute) PeriodEmbedder (class in jaxkan.models.utils)

phase (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.models.ActNet.ActLayer attribute) pi_init (jaxkan.models.RGAKAN.RGAKAN attribute)

R

RBFGrid (class in jaxkan.grids.RBFGrid) RBFLayer (class in jaxkan.layers.RBF) residual (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) RFFEmbedder (class in jaxkan.models.utils)

RGABlock (class in jaxkan.models.RGAKAN) RGAKAN (class in jaxkan.models.RGAKAN) rngs (jaxkan.layers.Chebyshev.ChebyshevLayer attribute) (jaxkan.layers.Legendre.LegendreLayer attribute) (jaxkan.layers.RBF.RBFLayer attribute) (jaxkan.layers.Sine.SineLayer attribute) (jaxkan.layers.Spline.BaseLayer attribute) (jaxkan.layers.Spline.SplineLayer attribute) (jaxkan.models.ActNet.ActLayer attribute) (jaxkan.models.ActNet.ActNet attribute)

S

SineBasis (jaxkan.models.RGAKAN.RGAKAN attribute) SineLayer (class in jaxkan.layers.Sine) solve_full_lstsq() (in module jaxkan.layers.utils)

solve_single_lstsq() (in module jaxkan.layers.utils) SplineGrid (class in jaxkan.grids.SplineGrid) SplineLayer (class in jaxkan.layers.Spline)

U

U (jaxkan.models.RGAKAN.RGAKAN attribute) update() (jaxkan.grids.BaseGrid.BaseGrid method) (jaxkan.grids.RBFGrid.RBFGrid method) (jaxkan.grids.SplineGrid.SplineGrid method) update_grid() (jaxkan.layers.Chebyshev.ChebyshevLayer method) (jaxkan.layers.Fourier.FourierLayer method) (jaxkan.layers.Legendre.LegendreLayer method) (jaxkan.layers.RBF.RBFLayer method) (jaxkan.layers.Sine.SineLayer method) (jaxkan.layers.Spline.BaseLayer method) (jaxkan.layers.Spline.SplineLayer method)

update_grids() (jaxkan.models.KAN.KAN method) update_rba_weights() (in module jaxkan.pikan.adaptive) use_exact (jaxkan.models.KKAN.ChebyshevEmbedding attribute) use_projections (jaxkan.models.ActNet.ActNet attribute)

V

v (jaxkan.layers.Dense.DenseLayer attribute)

V (jaxkan.models.RGAKAN.RGAKAN attribute)