Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | R | S | T | V | W | Z _ __call__() (pcax.core.RandomKeyGenerator method) (pcax.functional._BaseTransform method) (pcax.predictive_coding.Vode method) (pcax.utils.Mask method) __init__() (pcax.core.BaseParam method) (pcax.core.RandomKeyGenerator method) (pcax.functional._BaseTransform method) (pcax.predictive_coding.EnergyModule method) (pcax.predictive_coding.Ruleset method) (pcax.predictive_coding.Vode method) (pcax.utils.Mask method) (pcax.utils.Optim method) _BaseTransform (class in pcax.functional) _process_mask() (pcax.functional._BaseTransform static method) A ALL (pcax.predictive_coding.STATUS attribute) apply() (pcax.utils.Mask static method) apply_get_transformation() (pcax.predictive_coding.Ruleset method) apply_set_transformation() (pcax.predictive_coding.Ruleset method) apply_updates() (pcax.utils.Optim method) AvgPool2d (class in pcax.nn) B BaseModule (class in pcax.core) BaseParam (class in pcax.core) C ce_energy() (in module pcax.predictive_coding) clear() (pcax.utils.Optim method) clear_params() (pcax.predictive_coding.EnergyModule method) cond() (in module pcax.functional) Conv (class in pcax.nn) Conv2d (class in pcax.nn) D dtype (pcax.core.Param property) E energy() (pcax.predictive_coding.EnergyModule method) (pcax.predictive_coding.Vode method) EnergyModule (class in pcax.predictive_coding) EVAL (pcax.core.Module.MODE attribute) eval() (pcax.core.Module method) F filter() (pcax.predictive_coding.Ruleset method) G get() (in module pcax.core) (pcax.core.BaseParam method) (pcax.core.Param method) (pcax.core.ParamDict method) (pcax.predictive_coding.Vode method) H has() (pcax.utils.m method) has_not() (pcax.utils.m method) I INIT (pcax.predictive_coding.STATUS attribute) init() (pcax.utils.Optim method) is_eval (pcax.core.Module property) is_train (pcax.core.Module property) J jit() (in module pcax.functional) L Layer (class in pcax.nn) LayerNorm (class in pcax.nn) LayerParam (class in pcax.nn) LayerState (class in pcax.nn) Linear (class in pcax.nn) load_params() (in module pcax.utils) M m (class in pcax.utils) Mask (class in pcax.utils) MaxPool2d (class in pcax.nn) mode() (pcax.core.Module method) module pcax.core pcax.functional pcax.nn pcax.predictive_coding pcax.utils Module (class in pcax.core) Module.MODE (class in pcax.core) N ndim (pcax.core.Param property) NONE (pcax.core.Module.MODE attribute) (pcax.predictive_coding.STATUS attribute) O Optim (class in pcax.utils) P Param (class in pcax.core) ParamCache (class in pcax.core) ParamDict (class in pcax.core) pcax.core module pcax.functional module pcax.nn module pcax.predictive_coding module pcax.utils module R RandomKeyGenerator (class in pcax.core) Ruleset (class in pcax.predictive_coding) S save_params() (in module pcax.utils) scan() (in module pcax.functional) se_energy() (in module pcax.predictive_coding) seed() (pcax.core.RandomKeyGenerator method) set() (in module pcax.core) (pcax.core.BaseParam method) (pcax.core.Param method) (pcax.core.ParamDict method) (pcax.predictive_coding.Vode method) shape (pcax.core.Param property) shared() (in module pcax.nn) static() (in module pcax.core) STATUS (class in pcax.predictive_coding) status (pcax.predictive_coding.EnergyModule property) step() (in module pcax.utils) (pcax.utils.Optim method) submodules() (pcax.core.BaseModule method) switch() (in module pcax.functional) T TRAIN (pcax.core.Module.MODE attribute) train() (pcax.core.Module method) tree_apply() (in module pcax.core) tree_extract() (in module pcax.core) tree_inject() (in module pcax.core) tree_ref() (in module pcax.core) tree_unref() (in module pcax.core) V value_and_grad() (in module pcax.functional) vmap() (in module pcax.functional) Vode (class in pcax.predictive_coding) VodeParam (class in pcax.predictive_coding) VodeParam.Cache (class in pcax.predictive_coding) W while_loop() (in module pcax.functional) Z zero_energy() (in module pcax.predictive_coding)