Index

B

Buffer (class in madnis.integrator)

C

cat() (madnis.integrator.SampleBatch static method) ChannelData (class in madnis.integrator)

ChannelGroup (class in madnis.integrator) ChannelGrouping (class in madnis.integrator)

D

DiscreteMADE (class in madnis.nn)

DiscreteTransformer (class in madnis.nn) Distribution (class in madnis.nn)

F

filter() (madnis.integrator.Buffer method) Flow (class in madnis.nn) forward() (madnis.integrator.Integrand method) (madnis.nn.MaskedMLP method) (madnis.nn.MLP method) (madnis.nn.StackedMLP method)

I

init_with_grid() (madnis.nn.Flow method) initialize_integrator() (madnis.integrator.VegasPreTraining method) integral() (madnis.integrator.Integrator method) Integrand (class in madnis.integrator) integrate() (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method)

integration_metrics() (in module madnis.integrator) (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method) IntegrationMetrics (class in madnis.integrator) Integrator (class in madnis.integrator)

K

kl_divergence() (in module madnis.integrator)

kl_divergence_softclip() (in module madnis.integrator)

L

log_prob() (madnis.nn.DiscreteTransformer method) (madnis.nn.Distribution method) (madnis.nn.Flow method) (madnis.nn.MixedFlow method)

M

madnis.integrator module madnis.nn module map() (madnis.integrator.SampleBatch method) MaskedMLP (class in madnis.nn)

MixedFlow (class in madnis.nn) MLP (class in madnis.nn) module madnis.integrator madnis.nn multi_channel_loss() (in module madnis.integrator)

P

prob() (madnis.nn.Distribution method) (madnis.nn.Flow method)

R

rational_quadratic_spline() (in module madnis.nn) remap_channels() (madnis.integrator.Integrand method)

reset_parameters() (madnis.nn.StackedMLP method) rkl_divergence() (in module madnis.integrator)

S

sample() (madnis.integrator.Buffer method) (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method) (madnis.nn.DiscreteTransformer method) (madnis.nn.Distribution method) (madnis.nn.Flow method)

SampleBatch (class in madnis.integrator) split() (madnis.integrator.SampleBatch method) StackedMLP (class in madnis.nn) store() (madnis.integrator.Buffer method) stratified_variance() (in module madnis.integrator) stratified_variance_softclip() (in module madnis.integrator)

T

train() (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method) train_step() (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method)

TrainingStatus (class in madnis.integrator) transform() (madnis.nn.Flow method)

U

unconstrained_rational_quadratic_spline() (in module madnis.nn) unique_channel_count() (madnis.integrator.Integrand method) unweighting_metrics() (in module madnis.integrator) (madnis.integrator.Integrator method) (madnis.integrator.VegasPreTraining method)

UnweightingMetrics (class in madnis.integrator)

V

VegasPreTraining (class in madnis.integrator)