Index A | B | C | D | E | F | I | K | L | M | N | P | R | S | T | U A activations (neuralprocesses.coders.nn.UNet attribute) after_turn_layers (neuralprocesses.coders.nn.UNet attribute) agg (neuralprocesses.coders.deepset.DeepSet attribute) AssertNoAugmentation (class in neuralprocesses.coders.augment) AssertNoParallel (class in neuralprocesses.coders.shaping) AssertParallel (class in neuralprocesses.coders.shaping) Attention (class in neuralprocesses.coders.attention) Augment (class in neuralprocesses.coders.augment) B before_turn_layers (neuralprocesses.coders.nn.UNet attribute) C coder (neuralprocesses.coders.augment.Augment attribute) (neuralprocesses.coders.functional.FunctionalCoder attribute) (neuralprocesses.coders.mapdiag.MapDiagonal attribute) construct_agnp() (in module neuralprocesses.architectures.agnp) construct_climate_convgnp_mlp() (in module neuralprocesses.architectures.climate) construct_climate_convgnp_multires() (in module neuralprocesses.architectures.climate) construct_convgnp() (in module neuralprocesses.architectures.convgnp) construct_fullconvgnp() (in module neuralprocesses.architectures.fullconvgnp) construct_gnp() (in module neuralprocesses.architectures.gnp) Conv (class in neuralprocesses.coders.nn) conv_net (neuralprocesses.coders.nn.ConvNet attribute) ConvNet (class in neuralprocesses.coders.nn) current (neuralprocesses.coders.shaping.RestructureParallel attribute) D DeepSet (class in neuralprocesses.coders.deepset) DenseCovariancePSDTransform (class in neuralprocesses.coders.densecov) dim (neuralprocesses.coders.nn.Conv attribute) (neuralprocesses.coders.nn.ConvNet attribute) (neuralprocesses.coders.nn.UNet attribute) dim_head (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) disc (neuralprocesses.coders.functional.FunctionalCoder attribute) DivideByFirstChannel (class in neuralprocesses.coders.setconv.density) DivideByFirstHalf (class in neuralprocesses.coders.setconv.density) E encoder_x (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) encoder_xy (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) epsilon (neuralprocesses.coders.setconv.density.DivideByFirstChannel attribute) (neuralprocesses.coders.setconv.density.DivideByFirstHalf attribute) F FromDenseCovariance (class in neuralprocesses.coders.densecov) FunctionalCoder (class in neuralprocesses.coders.functional) Fuse (class in neuralprocesses.coders.fuse) I Identity (class in neuralprocesses.coders.shaping) InputsCoder (class in neuralprocesses.coders.inputs) K kernel (neuralprocesses.coders.nn.ConvNet attribute) kernels (neuralprocesses.coders.nn.UNet attribute) L layer1 (neuralprocesses.coders.nn.ResidualBlock attribute) layer2 (neuralprocesses.coders.nn.ResidualBlock attribute) layer_post (neuralprocesses.coders.nn.ResidualBlock attribute) Linear (class in neuralprocesses.coders.nn) ln1 (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) ln2 (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) log_scale (neuralprocesses.coders.setconv.setconv.SetConv attribute) M MapDiagonal (class in neuralprocesses.coders.mapdiag) mixer (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) MLP (class in neuralprocesses.coders.nn) mlp1 (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) mlp2 (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) module neuralprocesses.architectures.agnp neuralprocesses.architectures.climate neuralprocesses.architectures.convgnp neuralprocesses.architectures.fullconvgnp neuralprocesses.architectures.gnp neuralprocesses.coders.aggregate neuralprocesses.coders.attention neuralprocesses.coders.augment neuralprocesses.coders.copy neuralprocesses.coders.deepset neuralprocesses.coders.densecov neuralprocesses.coders.functional neuralprocesses.coders.fuse neuralprocesses.coders.inputs neuralprocesses.coders.mapdiag neuralprocesses.coders.nn neuralprocesses.coders.setconv.density neuralprocesses.coders.setconv.identity neuralprocesses.coders.setconv.setconv neuralprocesses.coders.shaping N n (neuralprocesses.coders.shaping.AssertParallel attribute) net (neuralprocesses.coders.nn.Conv attribute) (neuralprocesses.coders.nn.Linear attribute) (neuralprocesses.coders.nn.MLP attribute) neuralprocesses.architectures.agnp module neuralprocesses.architectures.climate module neuralprocesses.architectures.convgnp module neuralprocesses.architectures.fullconvgnp module neuralprocesses.architectures.gnp module neuralprocesses.coders.aggregate module neuralprocesses.coders.attention module neuralprocesses.coders.augment module neuralprocesses.coders.copy module neuralprocesses.coders.deepset module neuralprocesses.coders.densecov module neuralprocesses.coders.functional module neuralprocesses.coders.fuse module neuralprocesses.coders.inputs module neuralprocesses.coders.mapdiag module neuralprocesses.coders.nn module neuralprocesses.coders.setconv.density module neuralprocesses.coders.setconv.identity module neuralprocesses.coders.setconv.setconv module neuralprocesses.coders.shaping module new (neuralprocesses.coders.shaping.RestructureParallel attribute) num_halving_layers (neuralprocesses.coders.nn.ConvNet attribute) (neuralprocesses.coders.nn.UNet attribute) num_heads (neuralprocesses.coders.attention.Attention attribute) (neuralprocesses.coders.attention.SelfAttention attribute) P phi (neuralprocesses.coders.deepset.DeepSet attribute) PrependDensityChannel (class in neuralprocesses.coders.setconv.density) PrependIdentityChannel (class in neuralprocesses.coders.setconv.identity) PrependMultiDensityChannel (class in neuralprocesses.coders.setconv.density) R receptive_field (neuralprocesses.coders.nn.ConvNet attribute) (neuralprocesses.coders.nn.UNet attribute) receptive_fields (neuralprocesses.coders.nn.UNet attribute) RepeatForAggregateInputPairs (class in neuralprocesses.coders.aggregate) RepeatForAggregateInputs (class in neuralprocesses.coders.aggregate) ResidualBlock (class in neuralprocesses.coders.nn) RestructureParallel (class in neuralprocesses.coders.shaping) S SelectFromChannels (class in neuralprocesses.coders.aggregate) SelectFromDenseCovarianceChannels (class in neuralprocesses.coders.aggregate) SelfAttention (class in neuralprocesses.coders.attention) set_conv (neuralprocesses.coders.fuse.Fuse attribute) SetConv (class in neuralprocesses.coders.setconv.setconv) sizes (neuralprocesses.coders.shaping.Splitter attribute) Splitter (class in neuralprocesses.coders.shaping) SqueezeParallel (class in neuralprocesses.coders.shaping) strides (neuralprocesses.coders.nn.UNet attribute) T target (neuralprocesses.coders.functional.FunctionalCoder attribute) ToDenseCovariance (class in neuralprocesses.coders.densecov) U UNet (class in neuralprocesses.coders.nn)
