fastbnns package

Subpackages

• fastbnns.analysis package
  • Submodules
  • fastbnns.analysis.statistics module
    • compute_coverage()
  • Module contents
• fastbnns.bnn package
  • Submodules
  • fastbnns.bnn.base module
    • BNN
      • BNN.forward()
      • BNN.laplace_init()
      • BNN.named_parameters_tagged()
    • bnn_params_from_laplace()
  • fastbnns.bnn.inference module
    • AvgPool1d
      • AvgPool1d.functional()
      • AvgPool1d.n_dim
    • AvgPool2d
      • AvgPool2d.functional()
      • AvgPool2d.n_dim
    • AvgPool3d
      • AvgPool3d.functional()
      • AvgPool3d.n_dim
    • AvgPoolNd
      • AvgPoolNd.forward()
    • BasicPropagator
      • BasicPropagator.forward()
    • Conv1d
      • Conv1d.functional()
    • Conv2d
      • Conv2d.functional()
    • Conv3d
      • Conv3d.functional()
    • ConvNd
      • ConvNd.forward()
    • ConvTranspose1d
      • ConvTranspose1d.functional()
      • ConvTranspose1d.num_spatial_dims
    • ConvTranspose2d
      • ConvTranspose2d.functional()
      • ConvTranspose2d.num_spatial_dims
    • ConvTranspose3d
      • ConvTranspose3d.functional()
      • ConvTranspose3d.num_spatial_dims
    • ConvTransposeNd
      • ConvTransposeNd.forward()
    • JointUnscentedTransform
      • JointUnscentedTransform.forward()
    • LeakyReLUa
      • LeakyReLUa.forward()
    • Linear
      • Linear.forward()
      • Linear.functional()
    • MomentPropagator
      • MomentPropagator.forward()
    • MonteCarlo
      • MonteCarlo.forward()
    • ReLUa
      • ReLUa.forward()
    • UnscentedTransform
      • UnscentedTransform.forward()
  • fastbnns.bnn.losses module
    • BNNLoss
      • BNNLoss.beta
      • BNNLoss.kl_divergence
      • BNNLoss.neg_log_likelihood
    • ELBO
      • ELBO.beta
      • ELBO.forward()
      • ELBO.kl_divergence
      • ELBO.neg_log_likelihood
    • KLDivergence
      • KLDivergence.forward()
    • kl_divergence_sampled()
  • fastbnns.bnn.priors module
    • Distribution
      • Distribution.distribution
      • Distribution.log_prob()
      • Distribution.sample()
    • SpikeSlab
      • SpikeSlab.distribution
  • fastbnns.bnn.types module
    • MuVar
      • MuVar.add()
      • MuVar.add_()
      • MuVar.addcmul()
      • MuVar.apply()
      • MuVar.apply_method()
      • MuVar.div()
      • MuVar.mean()
      • MuVar.mul()
      • MuVar.numel()
      • MuVar.pow()
      • MuVar.sub()
      • MuVar.sub_()
    • implements()
    • muvar_avg_pool()
    • muvar_dropout()
  • fastbnns.bnn.wrappers module
    • BayesianModule
      • BayesianModule.compute_kl_divergence()
      • BayesianModule.forward()
      • BayesianModule.get_named_sampler()
      • BayesianModule.learn_var
      • BayesianModule.module
      • BayesianModule.module_map
      • BayesianModule.module_params
      • BayesianModule.moment_propagator
      • BayesianModule.reset_parameters()
      • BayesianModule.samplers
    • BayesianModuleBase
      • BayesianModuleBase.compute_kl_divergence()
      • BayesianModuleBase.forward()
      • BayesianModuleBase.learn_var
      • BayesianModuleBase.module
      • BayesianModuleBase.module_map
      • BayesianModuleBase.module_params
      • BayesianModuleBase.moment_propagator
      • BayesianModuleBase.scale_tform()
      • BayesianModuleBase.scale_tform_inv()
    • BroadcastModule
      • BroadcastModule.forward()
    • convert_to_bnn_()
    • convert_to_nn()
    • isolate_leaf_module_names()
    • select_default_propagator()
  • Module contents
• fastbnns.datasets package
  • Submodules
  • fastbnns.datasets.generic module
    • SimulatedData
  • fastbnns.datasets.polynomial module
    • Polynomial
  • Module contents
• fastbnns.models package
  • Submodules
  • fastbnns.models.activations module
    • InverseTransformSampling
      • InverseTransformSampling.alpha
      • InverseTransformSampling.forward()
    • scaled_sigmoid()
  • fastbnns.models.cnn module
    • CNN
      • CNN.forward()
  • fastbnns.models.lightning_wrappers module
  • fastbnns.models.mlp module
    • MLP
      • MLP.forward()
  • fastbnns.models.polynomial module
    • PolyModule
      • PolyModule.forward()
  • Module contents
• fastbnns.simulation package
  • Submodules
  • fastbnns.simulation.generators module
    • Generator
      • Generator.forward()
    • sample_simulator()
  • fastbnns.simulation.images module
  • fastbnns.simulation.observation module
    • NoiseTransform
      • NoiseTransform.forward()
    • add_read_noise()
    • sensor_noise()
  • fastbnns.simulation.polynomials module
    • polynomial()
  • Module contents
• fastbnns.utils package
  • Submodules
  • fastbnns.utils.torch_utils module
    • set_requires_grad_()
  • Module contents

Module contents