nifty8 package

Subpackages

• nifty8.domains package
  • Submodules
    • nifty8.domains.dof_space module
      • DOFSpace
    • nifty8.domains.domain module
      • Domain
    • nifty8.domains.gl_space module
      • GLSpace
    • nifty8.domains.hp_space module
      • HPSpace
    • nifty8.domains.lm_space module
      • LMSpace
    • nifty8.domains.power_space module
      • PowerSpace
    • nifty8.domains.rg_space module
      • RGSpace
    • nifty8.domains.structured_domain module
      • StructuredDomain
    • nifty8.domains.unstructured_domain module
      • UnstructuredDomain
  • Module contents
• nifty8.library package
  • Submodules
    • nifty8.library.adjust_variances module
      • do_adjust_variances()
      • make_adjust_variances_hamiltonian()
    • nifty8.library.correlated_fields module
      • CorrelatedFieldMaker
    • nifty8.library.correlated_fields_simple module
      • SimpleCorrelatedField()
    • nifty8.library.dynamic_operator module
      • dynamic_lightcone_operator()
      • dynamic_operator()
    • nifty8.library.light_cone_operator module
      • LightConeOperator
    • nifty8.library.los_response module
      • LOSResponse
      • apply_erf()
    • nifty8.library.nft module
      • FinuFFT()
      • Gridder
      • Nufft
    • nifty8.library.special_distributions module
      • GammaOperator
      • InverseGammaOperator
      • LaplaceOperator
      • LogInverseGammaOperator()
      • UniformOperator
    • nifty8.library.variational_models module
      • DiagonalSelector
      • FullCovarianceVI
      • GaussianEntropy
      • LowerTriangularInserter
      • MeanFieldVI
    • nifty8.library.wiener_filter_curvature module
      • WienerFilterCurvature()
  • Module contents
• nifty8.minimization package
  • Subpackages
    • nifty8.minimization.config package
      • Submodules
      • Module contents
  • Submodules
    • nifty8.minimization.conjugate_gradient module
      • ConjugateGradient
    • nifty8.minimization.descent_minimizers module
      • DescentMinimizer
      • L_BFGS
      • NewtonCG
      • RelaxedNewton
      • SteepestDescent
      • VL_BFGS
    • nifty8.minimization.energy module
      • Energy
    • nifty8.minimization.energy_adapter module
      • EnergyAdapter
      • StochasticEnergyAdapter
    • nifty8.minimization.iteration_controllers module
      • AbsDeltaEnergyController
      • DeltaEnergyController
      • EnergyHistory
      • GradInfNormController
      • GradientNormController
      • IterationController
      • StochasticAbsDeltaEnergyController
      • append_history()
    • nifty8.minimization.kl_energies module
      • SampledKLEnergy()
      • SampledKLEnergyClass
      • draw_samples()
    • nifty8.minimization.line_search module
      • LineEnergy
      • LineSearch
    • nifty8.minimization.minimizer module
      • Minimizer
    • nifty8.minimization.nonlinear_cg module
      • NonlinearCG
    • nifty8.minimization.optimize_kl module
      • optimize_kl()
    • nifty8.minimization.quadratic_energy module
      • QuadraticEnergy
    • nifty8.minimization.sample_list module
      • ResidualSampleList
      • SampleList
      • SampleListBase
    • nifty8.minimization.scipy_minimizer module
      • L_BFGS_B()
    • nifty8.minimization.stochastic_minimizer module
      • ADVIOptimizer
  • Module contents
• nifty8.operators package
  • Submodules
    • nifty8.operators.adder module
      • Adder
    • nifty8.operators.block_diagonal_operator module
      • BlockDiagonalOperator
    • nifty8.operators.chain_operator module
      • ChainOperator
    • nifty8.operators.contraction_operator module
      • ContractionOperator
      • IntegrationOperator()
    • nifty8.operators.convolution_operators module
      • FuncConvolutionOperator()
    • nifty8.operators.counting_operator module
      • CountingOperator
    • nifty8.operators.diagonal_operator module
      • DiagonalOperator
    • nifty8.operators.distributors module
      • DOFDistributor
      • PowerDistributor
    • nifty8.operators.domain_tuple_field_inserter module
      • DomainTupleFieldInserter
    • nifty8.operators.einsum module
      • LinearEinsum
      • MultiLinearEinsum
    • nifty8.operators.endomorphic_operator module
      • EndomorphicOperator
    • nifty8.operators.energy_operators module
      • AveragedEnergy
      • BernoulliEnergy
      • EnergyOperator
      • GaussianEnergy
      • InverseGammaEnergy
      • LikelihoodEnergyOperator
      • PoissonianEnergy
      • QuadraticFormOperator
      • Squared2NormOperator
      • StandardHamiltonian
      • StudentTEnergy
      • VariableCovarianceGaussianEnergy
    • nifty8.operators.field_zero_padder module
      • FieldZeroPadder
    • nifty8.operators.harmonic_operators module
      • FFTOperator
      • HarmonicSmoothingOperator()
      • HarmonicTransformOperator
      • HartleyOperator
      • SHTOperator
    • nifty8.operators.inversion_enabler module
      • InversionEnabler
    • nifty8.operators.jax_operator module
      • JaxLikelihoodEnergyOperator
      • JaxLinearOperator
      • JaxOperator
    • nifty8.operators.linear_interpolation module
      • LinearInterpolator
    • nifty8.operators.linear_operator module
      • LinearOperator
    • nifty8.operators.mask_operator module
      • MaskOperator
    • nifty8.operators.matrix_product_operator module
      • MatrixProductOperator
    • nifty8.operators.multifield2vector module
      • Multifield2Vector
    • nifty8.operators.normal_operators module
      • LognormalTransform()
      • NormalTransform()
    • nifty8.operators.operator module
      • Operator
      • func()
    • nifty8.operators.operator_adapter module
      • OperatorAdapter
    • nifty8.operators.outer_product_operator module
      • OuterProduct
    • nifty8.operators.partial_conjugate module
      • PartialConjugate
    • nifty8.operators.regridding_operator module
      • RegriddingOperator
    • nifty8.operators.sampling_enabler module
      • SamplingEnabler
    • nifty8.operators.sandwich_operator module
      • SandwichOperator
    • nifty8.operators.scaling_operator module
      • ScalingOperator
    • nifty8.operators.selection_operators module
      • SliceOperator
      • SplitOperator
    • nifty8.operators.simple_linear_operators module
      • ConjugationOperator
      • DomainChangerAndReshaper
      • FieldAdapter
      • GeometryRemover
      • Imaginizer
      • NullOperator
      • PartialExtractor
      • PrependKey
      • Realizer
      • VdotOperator
      • WeightApplier
      • ducktape()
    • nifty8.operators.simplify_for_const module
      • ConstCollector
      • ConstantEnergyOperator
      • ConstantLikelihoodEnergyOperator
      • ConstantOperator
      • InsertionOperator
    • nifty8.operators.sum_operator module
      • SumOperator
    • nifty8.operators.transpose_operator module
      • TransposeOperator
    • nifty8.operators.value_inserter module
      • ValueInserter
  • Module contents
• nifty8.re package
  • Subpackages
    • nifty8.re.refine package
      • Submodules
      • Module contents
  • Submodules
    • nifty8.re.conjugate_gradient module
      • CGResults
      • cg()
      • get_boundaries_intersections()
      • second_order_approx()
      • static_cg()
    • nifty8.re.correlated_field module
      • CorrelatedFieldMaker
      • get_fourier_mode_distributor()
      • hartley()
      • non_parametric_amplitude()
    • nifty8.re.disable_jax_control_flow module
      • cond()
      • fori_loop()
      • while_loop()
    • nifty8.re.energy_operators module
      • Categorical()
      • Gaussian()
      • Poissonian()
      • StudentT()
      • VariableCovarianceGaussian()
      • VariableCovarianceStudentT()
      • standard_t()
    • nifty8.re.field module
      • Field
      • dot()
      • matmul()
    • nifty8.re.forest_util module
      • ShapeWithDtype
      • assert_arithmetics()
      • common_type()
      • dot()
      • get_dtype()
      • get_map()
      • has_arithmetics()
      • map_forest()
      • map_forest_mean()
      • norm()
      • select()
      • shape()
      • size()
      • smap()
      • split()
      • stack()
      • unite()
      • unstack()
      • vdot()
      • where()
      • zeros_like()
    • nifty8.re.hmc module
      • AcceptedAndRejected
      • QP
      • Tree
      • add_single_qp_to_tree()
      • count_trailing_ones()
      • flip_momentum()
      • generate_hmc_acc_rej()
      • generate_nuts_tree()
      • is_euclidean_uturn()
      • iterative_build_tree()
      • leapfrog_step()
      • merge_trees()
      • sample_momentum_from_diagonal()
      • total_energy_of_qp()
      • tree_index_get()
      • tree_index_update()
    • nifty8.re.hmc_oo module
      • Chain
      • HMCChain
      • NUTSChain
    • nifty8.re.kl module
      • GeoMetricKL()
      • MetricKL()
      • SampleIter
      • cond_raise()
      • geometrically_sample_standard_hamiltonian()
      • sample_likelihood()
      • sample_standard_hamiltonian()
    • nifty8.re.kl_aux module
      • mean_hessp()
      • mean_metric()
      • mean_value_and_grad()
    • nifty8.re.lanczos module
      • lanczos_tridiag()
      • stochastic_logdet_from_lanczos()
      • stochastic_lq_logdet()
    • nifty8.re.likelihood module
      • Likelihood
      • StandardHamiltonian
    • nifty8.re.model module
      • AbstractModel
      • Initializer
      • Model
    • nifty8.re.num module
      • amend_unique()
      • amend_unique_()
      • unique()
    • nifty8.re.optimize module
      • OptimizeResults
      • minimize()
      • newton_cg()
      • trust_ncg()
    • nifty8.re.stats_distributions module
      • interpolator()
      • invgamma_invprior()
      • invgamma_prior()
      • laplace_prior()
      • lognormal_invprior()
      • lognormal_moments()
      • lognormal_prior()
      • normal_invprior()
      • normal_prior()
      • uniform_prior()
    • nifty8.re.structured_kernel_interpolation module
      • HarmonicSKI
      • ToeplitzSKI
      • interp_mat()
      • matmul_toeplitz()
    • nifty8.re.sugar module
      • doc_from()
      • ducktape()
      • ducktape_left()
      • interpolate()
      • is1d()
      • isiterable()
      • mean()
      • mean_and_std()
      • random_like()
      • sum_of_squares()
  • Module contents

Submodules

• nifty8.domain_tuple module
  • DomainTuple
    • DomainTuple.__init__()
    • DomainTuple.axes
    • DomainTuple.make()
    • DomainTuple.scalar_domain()
    • DomainTuple.scalar_weight()
    • DomainTuple.shape
    • DomainTuple.size
    • DomainTuple.total_volume()
• nifty8.ducc_dispatch module
  • fftn()
  • hartley()
  • ifftn()
  • nthreads()
  • set_nthreads()
  • vdot()
• nifty8.evidence_lower_bound module
  • estimate_evidence_lower_bound()
• nifty8.extra module
  • assert_allclose()
  • check_linear_operator()
  • check_operator()
  • minisanity()
• nifty8.field module
  • Field
    • Field.__init__()
    • Field.__repr__()
    • Field.__str__()
    • Field.all()
    • Field.any()
    • Field.cast_domain()
    • Field.conjugate()
    • Field.domain
    • Field.dtype
    • Field.extract()
    • Field.extract_part()
    • Field.flexible_addsub()
    • Field.from_random()
    • Field.from_raw()
    • Field.full()
    • Field.imag
    • Field.integrate()
    • Field.mean()
    • Field.norm()
    • Field.outer()
    • Field.prod()
    • Field.ptw()
    • Field.ptw_with_deriv()
    • Field.real
    • Field.s_all()
    • Field.s_any()
    • Field.s_integrate()
    • Field.s_mean()
    • Field.s_prod()
    • Field.s_std()
    • Field.s_sum()
    • Field.s_var()
    • Field.s_vdot()
    • Field.scalar()
    • Field.scalar_weight()
    • Field.scale()
    • Field.shape
    • Field.size
    • Field.std()
    • Field.sum()
    • Field.total_volume()
    • Field.unite()
    • Field.val
    • Field.val_rw()
    • Field.var()
    • Field.vdot()
    • Field.weight()
  • func()
• nifty8.linearization module
  • Linearization
    • Linearization.__init__()
    • Linearization.add_metric()
    • Linearization.conjugate()
    • Linearization.domain
    • Linearization.gradient
    • Linearization.imag
    • Linearization.integrate()
    • Linearization.jac
    • Linearization.make_const()
    • Linearization.make_const_empty_input()
    • Linearization.make_partial_var()
    • Linearization.make_var()
    • Linearization.metric
    • Linearization.new()
    • Linearization.outer()
    • Linearization.prepend_jac()
    • Linearization.ptw()
    • Linearization.real
    • Linearization.sum()
    • Linearization.target
    • Linearization.trivial_jac()
    • Linearization.val
    • Linearization.vdot()
    • Linearization.want_metric
    • Linearization.with_want_metric()
• nifty8.multi_domain module
  • MultiDomain
    • MultiDomain.__init__()
    • MultiDomain.domains()
    • MultiDomain.idx
    • MultiDomain.items()
    • MultiDomain.keys()
    • MultiDomain.make()
    • MultiDomain.size
    • MultiDomain.union()
    • MultiDomain.values()
• nifty8.multi_field module
  • MultiField
    • MultiField.__init__()
    • MultiField.clip()
    • MultiField.conjugate()
    • MultiField.domain
    • MultiField.dtype
    • MultiField.extract()
    • MultiField.extract_by_keys()
    • MultiField.extract_part()
    • MultiField.flexible_addsub()
    • MultiField.from_dict()
    • MultiField.from_random()
    • MultiField.from_raw()
    • MultiField.full()
    • MultiField.imag
    • MultiField.items()
    • MultiField.keys()
    • MultiField.norm()
    • MultiField.ptw()
    • MultiField.ptw_with_deriv()
    • MultiField.real
    • MultiField.s_all()
    • MultiField.s_any()
    • MultiField.s_sum()
    • MultiField.s_vdot()
    • MultiField.size
    • MultiField.to_dict()
    • MultiField.union()
    • MultiField.unite()
    • MultiField.val
    • MultiField.val_rw()
    • MultiField.values()
    • MultiField.vdot()
  • func()
• nifty8.nifty2jax module
  • convert()
  • shapewithdtype_from_domain()
  • spaces_to_axes()
  • wrap_nifty_call()
• nifty8.operator_spectrum module
  • operator_spectrum()
• nifty8.operator_tree_optimiser module
  • optimise_operator()
• nifty8.plot module
  • Plot
    • Plot.__init__()
    • Plot.add()
    • Plot.output()
  • plottable2D()
• nifty8.pointwise module
  • exponentiate()
  • exponentiate_j()
  • sigmoid_j()
  • softplus()
• nifty8.probing module
  • StatCalculator
    • StatCalculator.__init__()
    • StatCalculator.add()
    • StatCalculator.mean
    • StatCalculator.var
  • approximation2endo()
  • probe_diagonal()
  • probe_with_posterior_samples()
• nifty8.random module
  • Context
    • Context.__init__()
  • Random
    • Random.normal()
    • Random.pm1()
    • Random.uniform()
  • current_rng()
  • getState()
  • pop_sseq()
  • push_sseq()
  • push_sseq_from_seed()
  • setState()
  • spawn_sseq()
• nifty8.sugar module
  • PS_field()
  • abs()
  • absolute()
  • arctan()
  • calculate_position()
  • clip()
  • cos()
  • cosh()
  • create_harmonic_smoothing_operator()
  • create_power_operator()
  • density_estimator()
  • domain_union()
  • exec_time()
  • exp()
  • expm1()
  • exponentiate()
  • from_random()
  • full()
  • get_default_codomain()
  • get_signal_variance()
  • is_fieldlike()
  • is_likelihood_energy()
  • is_linearization()
  • is_operator()
  • log()
  • log10()
  • log1p()
  • makeDomain()
  • makeField()
  • makeOp()
  • plot_priorsamples()
  • power()
  • power_analyze()
  • reciprocal()
  • sigmoid()
  • sign()
  • sin()
  • sinc()
  • single_plot()
  • sinh()
  • softplus()
  • sqrt()
  • tan()
  • tanh()
  • unitstep()
• nifty8.utilities module
  • NiftyMeta
  • check_MPI_equality()
  • check_MPI_synced_random_state()
  • check_object_identity()
  • frozendict
    • frozendict.__init__()
    • frozendict.copy()
    • frozendict.dict_cls
  • get_slice_list()
  • indent()
  • infer_space()
  • iscomplextype()
  • issingleprec()
  • lognormal_moments()
  • memo()
  • my_lincomb()
  • my_lincomb_simple()
  • my_product()
  • my_sum()
  • parse_spaces()
  • safe_cast()
  • special_add_at()
  • value_reshaper()
• nifty8.version module

Module contents