Toggle navigation sidebar
Toggle in-page Table of Contents
API Reference
liesel.experimental
liesel.experimental.pymc
PyMCInterface
PyMCInterface.extract_position()
PyMCInterface.get_initial_state()
PyMCInterface.log_prob()
PyMCInterface.update_state()
liesel.liesel
ModelState
ModelState.clear()
ModelState.copy()
ModelState.fromkeys()
ModelState.get()
ModelState.items()
ModelState.keys()
ModelState.pop()
ModelState.popitem()
ModelState.setdefault()
ModelState.update()
ModelState.values()
liesel.liesel.Position
liesel.liesel.distreg
liesel.liesel.distreg.dist_reg_mcmc
liesel.liesel.distreg.tau2_gibbs_kernel
CopRegBuilder
DistRegBuilder
NPSmoothGroup
PSmoothGroup
SmoothGroup
liesel.liesel.goose
liesel.liesel.goose.get_log_prob
liesel.liesel.goose.get_position
liesel.liesel.goose.make_log_prob_fn
liesel.liesel.goose.make_update_state_fn
liesel.liesel.goose.update_state
GooseModel
liesel.liesel.model
liesel.liesel.model.load_model
liesel.liesel.model.save_model
Model
ModelBuilder
liesel.liesel.nodes
liesel.liesel.nodes.transform_parameter
Addition
AdditionCalculator
Bijector
BijectorCalculator
ColumnStack
ColumnStackCalculator
DesignMatrix
Hyperparameter
InverseLink
Node
NodeCalculator
NodeComponent
NodeDistribution
NodeGroup
NodeInputs
PIT
PITCalculator
Parameter
Predictor
RegressionCoef
Response
Smooth
SmoothCalculator
SmoothingParam
liesel.liesel.types
NodeState
liesel.liesel.viz
liesel.liesel.viz.plot_model
liesel.goose
liesel.goose.builder
EngineBuilder
liesel.goose.chain
Chain
EpochChain
EpochChainManager
ListChain
ListEpochChain
liesel.goose.da
liesel.goose.da.da_finalize
liesel.goose.da.da_init
liesel.goose.da.da_step
DAKernelState
liesel.goose.engine
liesel.goose.engine.stack_for_multi
Carry
Engine
KernelErrorLog
SamplingResult
SamplingResults
liesel.goose.epoch
EpochConfig
EpochManager
EpochState
EpochType
liesel.goose.gibbs
GibbsKernel
liesel.goose.hmc
HMCKernel
HMCKernelState
HMCTransitionInfo
liesel.goose.iwls
IWLSKernel
IWLSKernelState
liesel.goose.iwls_utils
liesel.goose.iwls_utils.mvn_log_prob
liesel.goose.iwls_utils.mvn_sample
liesel.goose.iwls_utils.solve
liesel.goose.kernel
DefaultTransitionInfo
DefaultTuningInfo
ModelMixin
TransitionMixin
TransitionOutcome
TuningMixin
TuningOutcome
WarmupOutcome
liesel.goose.kernel_sequence
KerSeqFinalizeWarmupOutput
KerSeqTransitionOutput
KerSeqTuningOutput
KernelSequence
liesel.goose.mh
liesel.goose.mh.mh_error_book
liesel.goose.mh.mh_step
liesel.goose.mh_kernel
MHKernel
MHProposal
liesel.goose.mm
liesel.goose.mm.tune_inv_mm_diag
liesel.goose.mm.tune_inv_mm_full
liesel.goose.models
DataClassModel
DictModel
liesel.goose.nuts
NUTSKernel
NUTSKernelState
NUTSTransitionInfo
liesel.goose.pytree
liesel.goose.pytree.as_strong_pytree
liesel.goose.pytree.concatenate_leaves
liesel.goose.pytree.register_dataclass_as_pytree
liesel.goose.pytree.slice_leaves
liesel.goose.pytree.split_and_transpose
liesel.goose.pytree.split_leaves
liesel.goose.pytree.squeeze_leaves
liesel.goose.pytree.stack_leaves
liesel.goose.rw
RWKernel
RWKernelState
liesel.goose.summary_m
liesel.goose.summary_m.add_hdi
liesel.goose.summary_m.add_num_effective
liesel.goose.summary_m.add_quantiles
liesel.goose.summary_m.add_rhat
liesel.goose.summary_m.adjust_dimensions
liesel.goose.summary_m.collect_param_dfs
liesel.goose.summary_m.collect_param_dicts
liesel.goose.summary_m.collect_subparam_dfs
liesel.goose.summary_m.collect_subparam_dicts
liesel.goose.summary_m.combine_chains
liesel.goose.summary_m.compute_hdi
liesel.goose.summary_m.compute_quantiles
liesel.goose.summary_m.compute_single_hdi
liesel.goose.summary_m.get_param_stats
liesel.goose.summary_m.get_subparam_stats
liesel.goose.summary_m.move_col_first
liesel.goose.summary_m.numpy_to_arviz
liesel.goose.summary_m.raise_chain_indices_error
liesel.goose.summary_m.raise_dimension_error
liesel.goose.summary_m.raise_param_indices_error
liesel.goose.summary_m.subparam_stats_to_df
liesel.goose.summary_m.summary
liesel.goose.summary_m.validate_chain_indices
liesel.goose.summary_m.validate_param_indices
liesel.goose.summary_m.validate_params
ErrorSummaryForOneCode
Summary
liesel.goose.summary_viz
liesel.goose.summary_viz.plot_cor
liesel.goose.summary_viz.plot_density
liesel.goose.summary_viz.plot_pairs
liesel.goose.summary_viz.plot_param
liesel.goose.summary_viz.plot_scatter
liesel.goose.summary_viz.plot_trace
liesel.goose.summary_viz.save_figure
liesel.goose.types
GeneratedQuantity
Kernel
ModelInterface
QuantityGenerator
TransitionInfo
TuningInfo
liesel.goose.warmup
liesel.goose.warmup.stan_epochs
liesel.tfp
liesel.tfp.jax
liesel.tfp.jax.bijectors
liesel.tfp.jax.distributions
liesel.tfp.numpy
liesel.tfp.numpy.bijectors
liesel.tfp.numpy.distributions
.rst
.pdf
Model.jaxify()
Model.jaxify()
#
Model.
jaxify
(
)
[source]
#
Enables JAX NumPy for all nodes in the model.
Return type
Model
