GitHub repository: pawel-czyz/pt-jax
Primary language: Python
Description: Minimal implementation of parallel tempering in JAX