User Guide
API Reference
msqrt()
squared_norm()
logger
plot_diagnostics()
plot_cornerplot()
DefaultNestedSampler
ApproximateNestedSampler
ExactNestedSampler
TerminationCondition
resample()
marginalise_static_from_U()
marginalise_dynamic_from_U()
marginalise_static()
marginalise_dynamic()
maximum_a_posteriori_point()
evaluate_map_estimate()
summary()
analytic_posterior_samples()
sample_evidence()
bruteforce_posterior_samples()
bruteforce_evidence()
save_pytree()
save_results()
load_pytree()
load_results()
PriorModelGen
PriorModelType
Model
Prior
InvalidPriorName
Bernoulli
Beta
Categorical
ForcedIdentifiability
Poisson
UnnormalisedDirichlet
Examples
jaxns.internals.linalg
Module Contents
Computes the matrix square-root using SVD, which is robust to poorly conditioned covariance matrices. Computes, M such that M @ M.T = A
A – [N,N] Square matrix to take square root of.
Returns: [N,N] matrix.