Skip to main content
Ctrl+K
 - Home
  • Quickstart

PPL INTEGRATION

  • Aesara
  • Numpyro
  • Oryx
  • PyMC
  • Tensorflow-Probability

HOW TO

  • Sample with multiple chains?
  • Use custom gradients?
  • Use non-JAX log-prob functions?
  • Build a Metropolis-Within-Gibbs sampler?
  • Sample from the word BlackJAX using BlackJAX?

LEARN BY EXAMPLE

  • The Sampling Book

API

  • API Reference
    • blackjax.adaptation
      • blackjax.adaptation.adjusted_mclmc_adaptation
      • blackjax.adaptation.base
      • blackjax.adaptation.chees_adaptation
      • blackjax.adaptation.mass_matrix
      • blackjax.adaptation.mclmc_adaptation
      • blackjax.adaptation.meads_adaptation
      • blackjax.adaptation.pathfinder_adaptation
      • blackjax.adaptation.step_size
      • blackjax.adaptation.window_adaptation
    • blackjax.diagnostics
    • blackjax.mcmc
      • blackjax.mcmc.adjusted_mclmc
      • blackjax.mcmc.adjusted_mclmc_dynamic
      • blackjax.mcmc.barker
      • blackjax.mcmc.diffusions
      • blackjax.mcmc.dynamic_hmc
      • blackjax.mcmc.elliptical_slice
      • blackjax.mcmc.ghmc
      • blackjax.mcmc.hmc
      • blackjax.mcmc.integrators
      • blackjax.mcmc.mala
      • blackjax.mcmc.marginal_latent_gaussian
      • blackjax.mcmc.mclmc
      • blackjax.mcmc.metrics
      • blackjax.mcmc.nuts
      • blackjax.mcmc.periodic_orbital
      • blackjax.mcmc.proposal
      • blackjax.mcmc.random_walk
      • blackjax.mcmc.rmhmc
      • blackjax.mcmc.termination
      • blackjax.mcmc.trajectory
    • blackjax.sgmcmc
      • blackjax.sgmcmc.csgld
      • blackjax.sgmcmc.diffusions
      • blackjax.sgmcmc.gradients
      • blackjax.sgmcmc.sghmc
      • blackjax.sgmcmc.sgld
      • blackjax.sgmcmc.sgnht
    • blackjax.smc
      • blackjax.smc.adaptive_tempered
      • blackjax.smc.base
      • blackjax.smc.ess
      • blackjax.smc.from_mcmc
      • blackjax.smc.inner_kernel_tuning
      • blackjax.smc.partial_posteriors_path
      • blackjax.smc.pretuning
      • blackjax.smc.resampling
      • blackjax.smc.solver
      • blackjax.smc.tempered
      • blackjax.smc.tuning
        • blackjax.smc.tuning.from_kernel_info
        • blackjax.smc.tuning.from_particles
      • blackjax.smc.waste_free
    • blackjax.vi
      • blackjax.vi.meanfield_vi
      • blackjax.vi.pathfinder
      • blackjax.vi.schrodinger_follmer
      • blackjax.vi.svgd
  • Bibliography

blackjax.mcmc.termination

Contents

  • Classes
  • Functions
  • Module Contents
    • IterativeUTurnState
      • IterativeUTurnState.momentum
      • IterativeUTurnState.momentum_sum
      • IterativeUTurnState.idx_min
      • IterativeUTurnState.idx_max
    • iterative_uturn_numpyro()

blackjax.mcmc.termination#

Classes#

IterativeUTurnState

Functions#

iterative_uturn_numpyro(is_turning)

Numpyro style dynamic U-Turn criterion.

Module Contents#

class IterativeUTurnState[source]#
momentum: blackjax.types.Array[source]#
momentum_sum: blackjax.types.Array[source]#
idx_min: int[source]#
idx_max: int[source]#
iterative_uturn_numpyro(is_turning: blackjax.mcmc.metrics.CheckTurning)[source]#

Numpyro style dynamic U-Turn criterion.

previous

blackjax.mcmc.rmhmc

next

blackjax.mcmc.trajectory

Contents
  • Classes
  • Functions
  • Module Contents
    • IterativeUTurnState
      • IterativeUTurnState.momentum
      • IterativeUTurnState.momentum_sum
      • IterativeUTurnState.idx_min
      • IterativeUTurnState.idx_max
    • iterative_uturn_numpyro()

By The Blackjax developers

© Copyright 2025, The Blackjax developers.