Index A | B | C | D | F | G | H | I | J | L | M | N | P | S | T | U | V A all_estimators() (in module bde.utils.discovery) apply() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) apply_reduced() (bde.ml.loss.GaussianNLLLoss method), [1] (bde.ml.loss.Loss method) (bde.ml.loss.LossMSE method) (bde.ml.loss.NLLLoss method) B BasicModule (class in bde.ml.models) bde module bde.ml module, [1] bde.ml.loss module bde.ml.models module bde.ml.training module bde.utils module, [1] bde.utils.configs module BDEEstimator (class in bde.ml.models), [1] bind() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) burn_in_loop() (bde.ml.models.BDEEstimator static method), [1] C clone() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) copy() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) D do_final_activation (bde.ml.models.FullyConnectedModule attribute) F fit() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) flax_training_loss_wrapper_classification() (in module bde.ml.loss) flax_training_loss_wrapper_regression() (in module bde.ml.loss) FullyConnectedEstimator (class in bde.ml.models) FullyConnectedModule (class in bde.ml.models) G GaussianNLLLoss (class in bde.ml.loss), [1] General (class in bde.utils.configs) get_metadata_routing() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) get_params() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) get_variable() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) H has_rng() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) has_variable() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) history_description() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) I init() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) init_dense_model() (in module bde.ml.models) init_dense_model_jitted() (in module bde.ml.models) init_inner_params() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) init_with_output() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) is_initializing() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) is_mutable_collection() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) J jitted_evaluation_for_a_metric() (in module bde.ml.training) jitted_evaluation_over_batch() (in module bde.ml.training) jitted_training() (in module bde.ml.training) jitted_training_epoch() (in module bde.ml.training) L layer_sizes (bde.ml.models.FullyConnectedModule attribute) lazy_init() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) load() (bde.ml.models.BDEEstimator class method), [1] (bde.ml.models.FullyConnectedEstimator class method) log_prior() (bde.ml.models.BDEEstimator method), [1] logdensity_for_batch() (bde.ml.models.BDEEstimator method), [1] Loss (class in bde.ml.loss) LossMSE (class in bde.ml.loss) M make_rng() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) mcmc_sampling() (bde.ml.models.BDEEstimator method), [1] module bde bde.ml, [1] bde.ml.loss bde.ml.models bde.ml.training bde.utils, [1] bde.utils.configs module_paths() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) N n_input_params (bde.ml.models.BasicModule attribute) (bde.ml.models.FullyConnectedModule attribute) n_output_params (bde.ml.models.BasicModule attribute) (bde.ml.models.FullyConnectedModule attribute) name (bde.ml.models.BasicModule attribute) (bde.ml.models.FullyConnectedModule attribute) NLLLoss (class in bde.ml.loss) P param() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) params (bde.ml.loss.GaussianNLLLoss attribute) (bde.ml.loss.NLLLoss attribute) parent (bde.ml.models.BasicModule attribute) (bde.ml.models.FullyConnectedModule attribute) path (bde.ml.models.BasicModule property) (bde.ml.models.FullyConnectedModule property) perturb() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) predict() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) predict_as_de() (bde.ml.models.BDEEstimator method), [1] predict_with_credibility_eti() (bde.ml.models.BDEEstimator method), [1] put_variable() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) S sample_from_samples() (bde.ml.models.BDEEstimator method), [1] save() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) scope (bde.ml.models.BasicModule attribute) (bde.ml.models.FullyConnectedModule attribute) SEED (bde.utils.configs.General attribute) set_fit_request() (bde.ml.models.BDEEstimator method), [1] set_params() (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) setup() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) sow() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) T tabulate() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) train_step() (in module bde.ml.training) tree_flatten() (bde.ml.loss.GaussianNLLLoss method), [1] (bde.ml.loss.Loss method) (bde.ml.loss.LossMSE method) (bde.ml.loss.NLLLoss method) (bde.ml.models.BasicModule method) (bde.ml.models.BDEEstimator method), [1] (bde.ml.models.FullyConnectedEstimator method) (bde.ml.models.FullyConnectedModule method) tree_unflatten() (bde.ml.loss.GaussianNLLLoss class method), [1] (bde.ml.loss.Loss class method) (bde.ml.loss.LossMSE class method) (bde.ml.loss.NLLLoss class method) (bde.ml.models.BasicModule class method) (bde.ml.models.BDEEstimator class method), [1] (bde.ml.models.FullyConnectedEstimator class method) (bde.ml.models.FullyConnectedModule class method) U unbind() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) V variable() (bde.ml.models.BasicModule method) (bde.ml.models.FullyConnectedModule method) variables (bde.ml.models.BasicModule property) (bde.ml.models.FullyConnectedModule property)
