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()