Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W A accelerations() (mimirax.adapters.jaccpot.JaccpotForceModel method) (mimirax.protocols.ForceModel method) (mimirax.testing.doubles.DirectSumGravity method) (mimirax.testing.doubles.SoftenedPointMassField method) adam() (in module mimirax.inference) (in module mimirax.inference.optimizers) AffineTransform (class in mimirax.parameters) also_fit (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) available() (mimirax._registry.Registry method) axes (mimirax.observables.ProjectedPositions attribute) (mimirax.observables.reference.ProjectedPositions attribute) B balanced() (mimirax.adapters.nornax.FoldedRollout class method) base_index (mimirax.adapters.nornax.NornaxRollout attribute) C centre_of_mass() (in module mimirax.parameters) centres (mimirax.observables.GaussianRadialBins attribute) (mimirax.observables.m2m.GaussianRadialBins attribute) checkpoint (mimirax.adapters.nornax.NornaxRollout attribute) checkpoint_substeps (mimirax.adapters.nornax.NornaxRollout attribute) chi_squared() (in module mimirax.likelihoods) (in module mimirax.likelihoods.gaussian) config (mimirax.adapters.odisseo.OdisseoForwardModel attribute) constrain() (in module mimirax.parameters) Constraint (class in mimirax.protocols) constraint_violation() (in module mimirax.parameters) covariance() (mimirax.inference.DataResampling method) (mimirax.inference.resampling.DataResampling method) D DataResampling (class in mimirax.inference) (class in mimirax.inference.resampling) decay_steps (mimirax.observables.m2m.TimeAverage attribute) (mimirax.observables.TimeAverage attribute) decorate() (mimirax._registry.Registry method) defect() (mimirax.parameters.FunctionConstraint method) (mimirax.parameters.TotalMassConstraint method) (mimirax.protocols.Constraint method) defect_fn (mimirax.parameters.FunctionConstraint attribute) degenerate_directions() (in module mimirax.diagnostics.degeneracy) DirectSumGravity (class in mimirax.testing.doubles) divergence() (mimirax.priors.EntropyPrior method) (mimirax.priors.reference.EntropyPrior method) dt (mimirax.adapters.nornax.NornaxRollout attribute) dynamical_masses() (mimirax.adapters.nornax.NornaxRollout method) E effective_parameters() (in module mimirax.diagnostics.degeneracy) effective_parameters_from_jacobian() (in module mimirax.diagnostics.degeneracy) EntropyPrior (class in mimirax.priors) (class in mimirax.priors.reference) eps (mimirax.adapters.nornax.NornaxRollout attribute) epsilon (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) eta (mimirax.adapters.nornax.NornaxRollout attribute) exact_gradient_sum_of_squares() (mimirax.testing.doubles.LinearForwardModel method) F final_state() (mimirax.adapters.nornax.NornaxRollout method) fisher_information() (in module mimirax.diagnostics.degeneracy) FitResult (class in mimirax.types) fold() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) (mimirax.protocols.FoldableObservable method) FoldableObservable (class in mimirax.protocols) FoldedRollout (class in mimirax.adapters.nornax) folding() (mimirax.adapters.nornax.NornaxRollout method) force (mimirax.adapters.nornax.NornaxRollout attribute) (mimirax.adapters.nornax.SingleRungMutualForce attribute) force_of_change() (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) ForceModel (class in mimirax.protocols) forward (mimirax.inference.InferenceProblem attribute) (mimirax.inference.objective.InferenceProblem attribute) forward() (mimirax.parameters.AffineTransform method) (mimirax.parameters.IdentityTransform method) (mimirax.parameters.LogTransform method) (mimirax.parameters.SoftplusTransform method) (mimirax.protocols.Reparameterization method) ForwardModel (class in mimirax.protocols) free_parameters() (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) from_preset() (mimirax.adapters.jaccpot.JaccpotForceModel class method) FunctionConstraint (class in mimirax.parameters) G GaussianLikelihood (class in mimirax.likelihoods) (class in mimirax.likelihoods.gaussian) GaussianPrior (class in mimirax.priors) (class in mimirax.priors.reference) GaussianRadialBins (class in mimirax.observables) (class in mimirax.observables.m2m) get() (mimirax._registry.Registry method) gravitational_constant (mimirax.testing.doubles.DirectSumGravity attribute) (mimirax.testing.doubles.SoftenedPointMassField attribute) H has_converged() (in module mimirax.diagnostics.convergence) HMC (class in mimirax.inference) (class in mimirax.inference.hmc) I IdentityObservable (class in mimirax.observables) (class in mimirax.observables.reference) IdentityTransform (class in mimirax.parameters) include_linear_term (mimirax.priors.EntropyPrior attribute) (mimirax.priors.reference.EntropyPrior attribute) InferenceProblem (class in mimirax.inference) (class in mimirax.inference.objective) initial() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) (mimirax.protocols.FoldableObservable method) inner (mimirax.observables.m2m.TimeAverage attribute) (mimirax.observables.TimeAverage attribute) inverse() (mimirax.parameters.AffineTransform method) (mimirax.parameters.IdentityTransform method) (mimirax.parameters.LogTransform method) (mimirax.parameters.SoftplusTransform method) (mimirax.protocols.Reparameterization method) J JaccpotForceModel (class in mimirax.adapters.jaccpot) K k_max (mimirax.adapters.nornax.NornaxRollout attribute) kernel (mimirax.observables.m2m.WeightedKernelSum attribute) (mimirax.observables.WeightedKernelSum attribute) kernel_values() (mimirax.observables.m2m.WeightedKernelSum method) (mimirax.observables.WeightedKernelSum method) key (mimirax.priors.EntropyPrior attribute) (mimirax.priors.reference.EntropyPrior attribute) keys (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) kind (mimirax._registry.Registry property) L L2Regularizer (class in mimirax.priors) (class in mimirax.priors.reference) l_curve_corner() (in module mimirax.diagnostics.degeneracy) learning_rate (mimirax.inference.MeanFieldVI attribute) (mimirax.inference.vi.MeanFieldVI attribute) level_accelerations() (mimirax.adapters.nornax.SingleRungMutualForce method) Likelihood (class in mimirax.protocols) likelihood (mimirax.inference.InferenceProblem attribute) (mimirax.inference.objective.InferenceProblem attribute) LinearForwardModel (class in mimirax.testing.doubles) log_abs_det_jacobian() (in module mimirax.parameters) (mimirax.parameters.AffineTransform method) (mimirax.parameters.IdentityTransform method) (mimirax.parameters.LogTransform method) (mimirax.parameters.SoftplusTransform method) (mimirax.protocols.Reparameterization method) log_density (mimirax.types.SampleResult attribute) log_likelihood() (mimirax.inference.InferenceProblem method) (mimirax.inference.objective.InferenceProblem method) log_posterior() (mimirax.inference.InferenceProblem method) (mimirax.inference.objective.InferenceProblem method) log_prior() (mimirax.inference.InferenceProblem method) (mimirax.inference.objective.InferenceProblem method) log_prob() (mimirax.likelihoods.gaussian.GaussianLikelihood method) (mimirax.likelihoods.GaussianLikelihood method) (mimirax.priors.EntropyPrior method) (mimirax.priors.GaussianPrior method) (mimirax.priors.L2Regularizer method) (mimirax.priors.reference.EntropyPrior method) (mimirax.priors.reference.GaussianPrior method) (mimirax.priors.reference.L2Regularizer method) (mimirax.protocols.Likelihood method) (mimirax.protocols.Prior method) LogTransform (class in mimirax.parameters) M made_to_measure() (in module mimirax.inference) (in module mimirax.inference.m2m) MadeToMeasure (class in mimirax.inference) (class in mimirax.inference.m2m) make_linear_problem() (in module mimirax.testing.doubles) make_observable() (in module mimirax.observables) masses (mimirax.adapters.nornax.NornaxRollout attribute) matrix (mimirax.testing.doubles.LinearForwardModel attribute) max_tree_depth (mimirax.inference.hmc.NUTS attribute) (mimirax.inference.NUTS attribute) mean (mimirax.priors.GaussianPrior attribute) (mimirax.priors.reference.GaussianPrior attribute) MeanFieldVI (class in mimirax.inference) (class in mimirax.inference.vi) mimirax._registry module mimirax.adapters.jaccpot module mimirax.adapters.nornax module mimirax.adapters.odisseo module mimirax.diagnostics.convergence module mimirax.diagnostics.degeneracy module mimirax.diagnostics.residuals module mimirax.inference module mimirax.inference.derivative_free module mimirax.inference.hmc module mimirax.inference.m2m module mimirax.inference.objective module mimirax.inference.optimizers module mimirax.inference.resampling module mimirax.inference.vi module mimirax.likelihoods module mimirax.likelihoods.gaussian module mimirax.observables module mimirax.observables.m2m module mimirax.observables.reference module mimirax.parameters module mimirax.priors module mimirax.priors.reference module mimirax.protocols module mimirax.testing.doubles module mimirax.types module minimize() (mimirax.inference.derivative_free.NelderMead method) (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) (mimirax.inference.NelderMead method) (mimirax.inference.OptaxOptimizer method) (mimirax.inference.optimizers.OptaxOptimizer method) (mimirax.protocols.Optimizer method) module mimirax._registry mimirax.adapters.jaccpot mimirax.adapters.nornax mimirax.adapters.odisseo mimirax.diagnostics.convergence mimirax.diagnostics.degeneracy mimirax.diagnostics.residuals mimirax.inference mimirax.inference.derivative_free mimirax.inference.hmc mimirax.inference.m2m mimirax.inference.objective mimirax.inference.optimizers mimirax.inference.resampling mimirax.inference.vi mimirax.likelihoods mimirax.likelihoods.gaussian mimirax.observables mimirax.observables.m2m mimirax.observables.reference mimirax.parameters mimirax.priors mimirax.priors.reference mimirax.protocols mimirax.testing.doubles mimirax.types moments (mimirax.observables.GaussianRadialBins attribute) (mimirax.observables.m2m.GaussianRadialBins attribute) mu (mimirax.priors.EntropyPrior attribute) (mimirax.priors.reference.EntropyPrior attribute) mutual_force() (mimirax.adapters.nornax.NornaxRollout method) N negative_log_posterior() (mimirax.inference.InferenceProblem method) (mimirax.inference.objective.InferenceProblem method) NelderMead (class in mimirax.inference) (class in mimirax.inference.derivative_free) net_momentum() (in module mimirax.parameters) normalized_residuals() (in module mimirax.diagnostics.residuals) NornaxRollout (class in mimirax.adapters.nornax) num_leapfrog_steps (mimirax.inference.HMC attribute) (mimirax.inference.hmc.HMC attribute) num_parameters() (in module mimirax.parameters) num_steps (mimirax.adapters.nornax.NornaxRollout attribute) (mimirax.inference.derivative_free.NelderMead attribute) (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) (mimirax.inference.MeanFieldVI attribute) (mimirax.inference.NelderMead attribute) (mimirax.inference.OptaxOptimizer attribute) (mimirax.inference.optimizers.OptaxOptimizer attribute) (mimirax.inference.vi.MeanFieldVI attribute) NUTS (class in mimirax.inference) (class in mimirax.inference.hmc) O objective_trace (mimirax.types.FitResult attribute) Observable (class in mimirax.protocols) observable (mimirax.adapters.nornax.FoldedRollout attribute) (mimirax.inference.InferenceProblem attribute) (mimirax.inference.objective.InferenceProblem attribute) observed (mimirax.inference.InferenceProblem attribute) (mimirax.inference.objective.InferenceProblem attribute) OdisseoForwardModel (class in mimirax.adapters.odisseo) offset (mimirax.testing.doubles.LinearForwardModel attribute) OptaxOptimizer (class in mimirax.inference) (class in mimirax.inference.optimizers) Optimizer (class in mimirax.protocols) optimizer (mimirax.inference.DataResampling attribute) (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) (mimirax.inference.OptaxOptimizer attribute) (mimirax.inference.optimizers.OptaxOptimizer attribute) (mimirax.inference.resampling.DataResampling attribute) P params (mimirax.adapters.odisseo.OdisseoForwardModel attribute) (mimirax.types.FitResult attribute) predict() (mimirax.inference.InferenceProblem method) (mimirax.inference.objective.InferenceProblem method) Prior (class in mimirax.protocols) priors (mimirax.inference.InferenceProblem attribute) (mimirax.inference.objective.InferenceProblem attribute) profile_likelihood() (in module mimirax.diagnostics.degeneracy) ProjectedPositions (class in mimirax.observables) (class in mimirax.observables.reference) Q quadratic_penalty() (in module mimirax.parameters) R ravel() (in module mimirax.parameters) reassign_rungs (mimirax.adapters.nornax.NornaxRollout attribute) rebuild_every (mimirax.adapters.nornax.NornaxRollout attribute) rebuild_fn (mimirax.adapters.nornax.NornaxRollout attribute) record (mimirax.adapters.nornax.NornaxRollout attribute) reference (mimirax.priors.EntropyPrior attribute) (mimirax.priors.reference.EntropyPrior attribute) refine (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) refine_steps (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) register() (mimirax._registry.Registry method) Registry (class in mimirax._registry) relative_change() (in module mimirax.diagnostics.convergence) remove_centre_of_mass() (in module mimirax.parameters) remove_net_momentum() (in module mimirax.parameters) Reparameterization (class in mimirax.protocols) residuals() (in module mimirax.diagnostics.residuals) result() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) (mimirax.protocols.FoldableObservable method) rms_residual() (in module mimirax.diagnostics.residuals) rollout (mimirax.adapters.nornax.FoldedRollout attribute) rollout() (mimirax.adapters.nornax.NornaxRollout method) S sample() (mimirax.inference.DataResampling method) (mimirax.inference.HMC method) (mimirax.inference.hmc.HMC method) (mimirax.inference.hmc.NUTS method) (mimirax.inference.MeanFieldVI method) (mimirax.inference.NUTS method) (mimirax.inference.resampling.DataResampling method) (mimirax.inference.vi.MeanFieldVI method) (mimirax.protocols.Sampler method) Sampler (class in mimirax.protocols) SampleResult (class in mimirax.types) samples (mimirax.types.SampleResult attribute) scale (mimirax.parameters.AffineTransform attribute) segments (mimirax.adapters.nornax.FoldedRollout attribute) self_consistent() (mimirax.adapters.nornax.NornaxRollout class method) sequential (mimirax.inference.DataResampling attribute) (mimirax.inference.resampling.DataResampling attribute) shift (mimirax.parameters.AffineTransform attribute) sigma (mimirax.likelihoods.gaussian.GaussianLikelihood attribute) (mimirax.likelihoods.GaussianLikelihood attribute) (mimirax.priors.GaussianPrior attribute) (mimirax.priors.reference.GaussianPrior attribute) SingleRungMutualForce (class in mimirax.adapters.nornax) SoftenedPointMassField (class in mimirax.testing.doubles) softening (mimirax.testing.doubles.DirectSumGravity attribute) (mimirax.testing.doubles.SoftenedPointMassField attribute) SoftplusTransform (class in mimirax.parameters) solver (mimirax.adapters.jaccpot.JaccpotForceModel attribute) source_masses (mimirax.testing.doubles.SoftenedPointMassField attribute) sources (mimirax.testing.doubles.SoftenedPointMassField attribute) step_size (mimirax.inference.HMC attribute) (mimirax.inference.hmc.HMC attribute) (mimirax.inference.hmc.NUTS attribute) (mimirax.inference.NUTS attribute) step_values() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) T tidal_tensor() (mimirax.testing.doubles.SoftenedPointMassField method) TimeAverage (class in mimirax.observables) (class in mimirax.observables.m2m) to_constrained() (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) to_unconstrained() (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) total (mimirax.parameters.TotalMassConstraint attribute) TotalMassConstraint (class in mimirax.parameters) tracer() (mimirax.adapters.nornax.NornaxRollout class method) transform (mimirax.inference.m2m.MadeToMeasure attribute) (mimirax.inference.MadeToMeasure attribute) U unconstrain() (in module mimirax.parameters) V value() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) (mimirax.protocols.FoldableObservable method) W weight (mimirax.priors.L2Regularizer attribute) (mimirax.priors.reference.L2Regularizer attribute) weight_keys() (mimirax.inference.m2m.MadeToMeasure method) (mimirax.inference.MadeToMeasure method) WeightedKernelSum (class in mimirax.observables) (class in mimirax.observables.m2m) weights() (mimirax.observables.m2m.TimeAverage method) (mimirax.observables.TimeAverage method) weights_are_masses (mimirax.adapters.nornax.NornaxRollout attribute) weights_key (mimirax.observables.m2m.WeightedKernelSum attribute) (mimirax.observables.WeightedKernelSum attribute) width (mimirax.observables.GaussianRadialBins attribute) (mimirax.observables.m2m.GaussianRadialBins attribute)