elisa.infer.samplers

• elisa.infer.samplers.blackjax
  • elisa.infer.samplers.blackjax.nuts
    • BlackJAXNUTSState
      • BlackJAXNUTSState.i
      • BlackJAXNUTSState.z
      • BlackJAXNUTSState.z_grad
      • BlackJAXNUTSState.potential_energy
      • BlackJAXNUTSState.energy
      • BlackJAXNUTSState.r
      • BlackJAXNUTSState.num_steps
      • BlackJAXNUTSState.tree_depth
      • BlackJAXNUTSState.accept_prob
      • BlackJAXNUTSState.mean_accept_prob
      • BlackJAXNUTSState.diverging
      • BlackJAXNUTSState.step_size
      • BlackJAXNUTSState.inverse_mass_matrix
      • BlackJAXNUTSState.adapt_state
      • BlackJAXNUTSState.rng_key
    • BlackJAXNUTS
      • BlackJAXNUTS.postprocess_fn()
      • BlackJAXNUTS.init()
      • BlackJAXNUTS.sample()
      • BlackJAXNUTS.sample_field
      • BlackJAXNUTS.default_fields
      • BlackJAXNUTS.get_diagnostics_str()
• elisa.infer.samplers.ensemble
  • elisa.infer.samplers.ensemble.base
    • EnsembleSamplerState
      • EnsembleSamplerState.coords
      • EnsembleSamplerState.random_state
    • EnsembleSampler
      • EnsembleSampler.run()
      • EnsembleSampler.get_sampling_fn()
      • EnsembleSampler.get_random_state()
  • elisa.infer.samplers.ensemble.emcee
    • EmceeSampler
      • EmceeSampler.get_sampling_fn()
      • EmceeSampler.get_random_state()
  • elisa.infer.samplers.ensemble.numpyro
    • NumpyroEnsembleSampler
      • NumpyroEnsembleSampler.sample_field
      • NumpyroEnsembleSampler.postprocess_fn()
      • NumpyroEnsembleSampler.init()
      • NumpyroEnsembleSampler.sample()
    • NumPyroAIES
      • NumPyroAIES.get_diagnostics_str()
    • NumPyroESS
  • elisa.infer.samplers.ensemble.zeus
    • ZeusSampler
      • ZeusSampler.get_sampling_fn()
      • ZeusSampler.get_random_state()
    • DifferentialMove
      • DifferentialMove.get_direction()
• elisa.infer.samplers.ns
  • elisa.infer.samplers.ns.jaxns
    • JAXNSSampler
      • JAXNSSampler.run()
      • JAXNSSampler.get_samples()
      • JAXNSSampler.get_weighted_samples()
      • JAXNSSampler.print_summary()
      • JAXNSSampler.diagnostics()
  • elisa.infer.samplers.ns.nautilus
    • NautilusSampler
      • NautilusSampler.run()
      • NautilusSampler.ess
      • NautilusSampler.lnZ
  • elisa.infer.samplers.ns.ultranest
    • UltraNestSampler
      • UltraNestSampler.run()
      • UltraNestSampler.print_results()
      • UltraNestSampler.ess
      • UltraNestSampler.lnZ

• elisa.infer.samplers.util
  • ModelInfo
    • ModelInfo.ndim
    • ModelInfo.init
    • ModelInfo.init_ravel
    • ModelInfo.unravel
    • ModelInfo.log_prob_fn
    • ModelInfo.postprocess_fn
    • ModelInfo.params_names
    • ModelInfo.params_dtype
    • ModelInfo.deterministic_names
    • ModelInfo.deterministic_dtype
  • get_model_info()
  • ravel_params_names()
  • UniformReparam
  • uniform_reparam_transform()
  • uniform_reparam_model()