Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | Q | R | S | T | U | V | W | Z _ __init__() (braincore.mixin.AlignPost method) (braincore.mixin.Batching method) (braincore.mixin.BindCondData method) (braincore.mixin.Delayed method) (braincore.mixin.DelayedInit method) (braincore.mixin.JointMode method) (braincore.mixin.Mixin method) (braincore.mixin.Mode method) (braincore.mixin.Training method) (braincore.mixin.UpdateReturn method) (braincore.util.DictManager method) (braincore.util.DotDict method) (braincore.util.IdMemScaling method) (braincore.util.MemScaling method) A AlignPost (class in braincore.mixin) AllOfTypes (in module braincore.mixin) append() (braincore.visible_module_list method) Arctan (class in braincore.surrogate) arctan() (in module braincore.surrogate) as_numpy() (in module braincore.math) assign_values() (braincore.StateDictManager method) at() (braincore.Delay method) B Batching (class in braincore.mixin) bernoulli (class in braincore.random) bernoulli() (braincore.random.RandomState method) beta (class in braincore.random) beta() (braincore.random.RandomState method) BindCondData (class in braincore.mixin) binomial (class in braincore.random) binomial() (braincore.random.RandomState method) braincore module braincore.math module braincore.mixin module braincore.random module braincore.surrogate module braincore.transform module braincore.util module C call_order (class in braincore) categorical (class in braincore.random) categorical() (braincore.random.RandomState method) check_state_value_tree (class in braincore) chisquare (class in braincore.random) chisquare() (braincore.random.RandomState method) choice (class in braincore.random) choice() (braincore.random.RandomState method) clear_buffer_memory() (in module braincore.util) clear_cache() (braincore.transform.StatefulFunction method) clip_by_norm() (in module braincore.math) collect_values() (braincore.StateDictManager method) compile_and_get_states_by_static_args() (braincore.transform.StatefulFunction method) cond() (in module braincore.transform) D default_rng (class in braincore.random) Delay (class in braincore) DelayAccess (class in braincore) Delayed (class in braincore.mixin) DelayedInit (class in braincore.mixin) DictManager (class in braincore.util) dirichlet (class in braincore.random) dirichlet() (braincore.random.RandomState method) DotDict (class in braincore.util) Dynamics (class in braincore) E ein_rearrange() (in module braincore.math) ein_reduce() (in module braincore.math) ein_repeat() (in module braincore.math) ein_shape() (in module braincore.math) ERF (class in braincore.surrogate) erf() (in module braincore.surrogate) exponential (class in braincore.random) exponential() (braincore.random.RandomState method) exprel() (in module braincore.math) extend() (braincore.visible_module_list method) F f (class in braincore.random) f() (braincore.random.RandomState method) flatten() (in module braincore.math) for_loop() (in module braincore.transform) from_numpy() (in module braincore.math) FullProjAlignPost (class in braincore) FullProjAlignPostMg (class in braincore) FullProjAlignPreDS (class in braincore) FullProjAlignPreDSMg (class in braincore) FullProjAlignPreSD (class in braincore) FullProjAlignPreSDMg (class in braincore) FullProjDelta (class in braincore) G gamma (class in braincore.random) gamma() (braincore.random.RandomState method) gaussian_grad() (in module braincore.surrogate) GaussianGrad (class in braincore.surrogate) geometric (class in braincore.random) geometric() (braincore.random.RandomState method) get_arg_cache_key() (braincore.transform.StatefulFunction method) get_dtype() (in module braincore.math) get_jaxpr() (braincore.transform.StatefulFunction method) get_out_shapes() (braincore.transform.StatefulFunction method) get_out_treedef() (braincore.transform.StatefulFunction method) get_read_states() (braincore.transform.StatefulFunction method) get_return_delay_at() (braincore.Dynamics method) get_states() (braincore.transform.StatefulFunction method) get_write_states() (braincore.transform.StatefulFunction method) grad() (in module braincore.transform) gumbel (class in braincore.random) gumbel() (braincore.random.RandomState method) H HalfProjAlignPost (class in braincore) HalfProjAlignPostMg (class in braincore) HalfProjDelta (class in braincore) hessian() (in module braincore.transform) hypergeometric (class in braincore.random) hypergeometric() (braincore.random.RandomState method) I IdMemScaling (class in braincore.util) ifelse() (in module braincore.transform) init_state() (braincore.Delay method) (braincore.Module method) init_states (class in braincore) inv_square_grad() (in module braincore.surrogate) InvSquareGrad (class in braincore.surrogate) is_float() (in module braincore.math) is_instance_eval() (in module braincore.util) is_int() (in module braincore.math) J jacfwd() (in module braincore.transform) jacobian() (in module braincore.transform) jacrev() (in module braincore.transform) jaxpr_call() (braincore.transform.StatefulFunction method) jaxpr_call_auto() (braincore.transform.StatefulFunction method) jit() (in module braincore.transform) jit_error (class in braincore.transform) JointMode (class in braincore.mixin) L laplace (class in braincore.random) laplace() (braincore.random.RandomState method) leaky_relu() (in module braincore.surrogate) LeakyRelu (class in braincore.surrogate) load_state() (braincore.Module method) load_states (class in braincore) log_tailed_relu() (in module braincore.surrogate) loggamma (class in braincore.random) loggamma() (braincore.random.RandomState method) logistic (class in braincore.random) logistic() (braincore.random.RandomState method) lognormal (class in braincore.random) lognormal() (braincore.random.RandomState method) logseries (class in braincore.random) logseries() (braincore.random.RandomState method) LogTailedRelu (class in braincore.surrogate) LongTermState (class in braincore) M make_jaxpr (class in braincore.transform) make_jaxpr() (braincore.transform.StatefulFunction method) maxwell (class in braincore.random) maxwell() (braincore.random.RandomState method) MemScaling (class in braincore.util) Mixin (class in braincore.mixin) mode (braincore.Module property) Mode (class in braincore.mixin) module braincore braincore.math braincore.mixin braincore.random braincore.surrogate braincore.transform braincore.util Module (class in braincore) ModuleGroup (class in braincore) multi_gaussian_grad() (in module braincore.surrogate) MultiGaussianGrad (class in braincore.surrogate) multinomial (class in braincore.random) multinomial() (braincore.random.RandomState method) multivariate_normal (class in braincore.random) multivariate_normal() (braincore.random.RandomState method) N name (braincore.Module property) negative_binomial (class in braincore.random) negative_binomial() (braincore.random.RandomState method) nodes() (braincore.Module method) noncentral_chisquare (class in braincore.random) noncentral_chisquare() (braincore.random.RandomState method) noncentral_f (class in braincore.random) noncentral_f() (braincore.random.RandomState method) nonzero_sign_log() (in module braincore.surrogate) NonzeroSignLog (class in braincore.surrogate) normal (class in braincore.random) normal() (braincore.random.RandomState method) not_instance_eval() (in module braincore.util) O OneOfTypes (in module braincore.mixin) orthogonal (class in braincore.random) orthogonal() (braincore.random.RandomState method) P ParamState (class in braincore) pareto (class in braincore.random) pareto() (braincore.random.RandomState method) permutation (class in braincore.random) permutation() (braincore.random.RandomState method) piecewise_exp() (in module braincore.surrogate) piecewise_leaky_relu() (in module braincore.surrogate) piecewise_quadratic() (in module braincore.surrogate) PiecewiseExp (class in braincore.surrogate) PiecewiseLeakyRelu (class in braincore.surrogate) PiecewiseQuadratic (class in braincore.surrogate) poisson (class in braincore.random) poisson() (braincore.random.RandomState method) power (class in braincore.random) power() (braincore.random.RandomState method) ProgressBar (class in braincore.transform) Projection (class in braincore) Q q_pseudo_spike() (in module braincore.surrogate) QPseudoSpike (class in braincore.surrogate) R rand (class in braincore.random) rand() (braincore.random.RandomState method) rand_like (class in braincore.random) rand_like() (braincore.random.RandomState method) randint (class in braincore.random) randint() (braincore.random.RandomState method) randint_like (class in braincore.random) randint_like() (braincore.random.RandomState method) randn (class in braincore.random) randn() (braincore.random.RandomState method) randn_like (class in braincore.random) randn_like() (braincore.random.RandomState method) random (class in braincore.random) random() (braincore.random.RandomState method) random_integers (class in braincore.random) random_integers() (braincore.random.RandomState method) random_sample (class in braincore.random) random_sample() (braincore.random.RandomState method) RandomState (class in braincore.random) ranf (class in braincore.random) ranf() (braincore.random.RandomState method) rayleigh (class in braincore.random) rayleigh() (braincore.random.RandomState method) register_entry() (braincore.Delay method) register_return_delay() (braincore.Dynamics method) relu_grad() (in module braincore.surrogate) ReluGrad (class in braincore.surrogate) remove_diag() (in module braincore.math) retrieve() (braincore.Delay method) S S2NN (class in braincore.surrogate) s2nn() (in module braincore.surrogate) sample (class in braincore.random) sample() (braincore.random.RandomState method) save_state() (braincore.Module method) save_states (class in braincore) scan() (in module braincore.transform) seed (class in braincore.random) seed() (braincore.random.RandomState method) Sequential (class in braincore) ShortTermState (class in braincore) shuffle (class in braincore.random) shuffle() (braincore.random.RandomState method) Sigmoid (class in braincore.surrogate) sigmoid() (in module braincore.surrogate) slayer_grad() (in module braincore.surrogate) SlayerGrad (class in braincore.surrogate) soft_sign() (in module braincore.surrogate) SoftSign (class in braincore.surrogate) source_info (braincore.State property) split() (braincore.StateDictManager method) split_key (class in braincore.random) split_key() (braincore.random.RandomState method) split_keys (class in braincore.random) split_keys() (braincore.random.RandomState method) split_values() (braincore.StateDictManager method) squarewave_fourier_series() (in module braincore.surrogate) SquarewaveFourierSeries (class in braincore.surrogate) standard_cauchy (class in braincore.random) standard_cauchy() (braincore.random.RandomState method) standard_exponential (class in braincore.random) standard_exponential() (braincore.random.RandomState method) standard_gamma (class in braincore.random) standard_gamma() (braincore.random.RandomState method) standard_normal (class in braincore.random) standard_normal() (braincore.random.RandomState method) standard_t (class in braincore.random) standard_t() (braincore.random.RandomState method) State (class in braincore) StateDictManager (class in braincore) StatefulFunction (class in braincore.transform) states() (braincore.Module method) surrogate_fun() (braincore.surrogate.Arctan method) (braincore.surrogate.ERF method) (braincore.surrogate.LeakyRelu method) (braincore.surrogate.LogTailedRelu method) (braincore.surrogate.NonzeroSignLog method) (braincore.surrogate.PiecewiseExp method) (braincore.surrogate.PiecewiseLeakyRelu method) (braincore.surrogate.PiecewiseQuadratic method) (braincore.surrogate.QPseudoSpike method) (braincore.surrogate.S2NN method) (braincore.surrogate.Sigmoid method) (braincore.surrogate.SoftSign method) (braincore.surrogate.SquarewaveFourierSeries method) surrogate_grad() (braincore.surrogate.Arctan method) (braincore.surrogate.ERF method) (braincore.surrogate.GaussianGrad method) (braincore.surrogate.InvSquareGrad method) (braincore.surrogate.LeakyRelu method) (braincore.surrogate.LogTailedRelu method) (braincore.surrogate.MultiGaussianGrad method) (braincore.surrogate.NonzeroSignLog method) (braincore.surrogate.PiecewiseExp method) (braincore.surrogate.PiecewiseLeakyRelu method) (braincore.surrogate.PiecewiseQuadratic method) (braincore.surrogate.QPseudoSpike method) (braincore.surrogate.ReluGrad method) (braincore.surrogate.S2NN method) (braincore.surrogate.Sigmoid method) (braincore.surrogate.SlayerGrad method) (braincore.surrogate.SoftSign method) (braincore.surrogate.SquarewaveFourierSeries method) switch() (in module braincore.transform) T t (class in braincore.random) t() (braincore.random.RandomState method) to_dict_values() (braincore.StateDictManager method) Training (class in braincore.mixin) tree_flatten() (braincore.State method) tree_ones_like() (in module braincore.math) tree_unflatten() (braincore.State class method) tree_zeros_like() (in module braincore.math) triangular (class in braincore.random) triangular() (braincore.random.RandomState method) truncated_normal (class in braincore.random) truncated_normal() (braincore.random.RandomState method) U unflatten() (in module braincore.math) uniform (class in braincore.random) uniform() (braincore.random.RandomState method) unique_name() (in module braincore.util) update() (braincore.Delay method) (braincore.DelayAccess method) (braincore.FullProjAlignPost method) (braincore.FullProjAlignPostMg method) (braincore.FullProjAlignPreDS method) (braincore.FullProjAlignPreDSMg method) (braincore.FullProjAlignPreSD method) (braincore.FullProjAlignPreSDMg method) (braincore.FullProjDelta method) (braincore.HalfProjAlignPost method) (braincore.HalfProjAlignPostMg method) (braincore.HalfProjDelta method) (braincore.Module method) (braincore.ModuleGroup method) (braincore.Projection method) (braincore.Sequential method) (braincore.VanillaProj method) (braincore.visible_module_dict method) update_return() (braincore.Dynamics method) (braincore.Sequential method) update_return_info() (braincore.Dynamics method) (braincore.Sequential method) UpdateReturn (class in braincore.mixin) V value (braincore.State property) VanillaProj (class in braincore) varshape (braincore.Dynamics property) vector_grad() (in module braincore.transform) visible_module_dict (class in braincore) visible_module_list (class in braincore) visible_state_dict (class in braincore) vonmises (class in braincore.random) vonmises() (braincore.random.RandomState method) W wald (class in braincore.random) wald() (braincore.random.RandomState method) weibull (class in braincore.random) weibull() (braincore.random.RandomState method) weibull_min (class in braincore.random) weibull_min() (braincore.random.RandomState method) while_loop() (in module braincore.transform) Z zipf (class in braincore.random) zipf() (braincore.random.RandomState method)
