jax-sgmc
latest
Getting Started
Installation
Quickstart
Reference Documentation
Data Loading
Compute Potential from Likelihood
Saving of Samples
Setup Schedulers
Setup Custom Solver
Advanced Topics
Extend Adapted Quantities
Extending Schedulers
Examples
Image Classification on CIFAR-10
API Documentation
jax_sgmc
jax_sgmc.adaption
jax_sgmc.alias
jax_sgmc.data
jax_sgmc.integrator
jax_sgmc.io
jax_sgmc.potential
jax_sgmc.scheduler
jax_sgmc.solver
jax_sgmc.util
jax-sgmc
jax_sgmc
Edit on GitHub
jax_sgmc
jax_sgmc
jax_sgmc.adaption
Adaption Strategies
Developer Information
jax_sgmc.alias
Solvers
jax_sgmc.data
jax_sgmc.data.core
jax_sgmc.data.numpy_loader
jax_sgmc.data.tensorflow_loader
jax_sgmc.data.hdf5_loader
jax_sgmc.integrator
Overview
Integrators
Integrator States
Utility
jax_sgmc.io
Data Collectors
Saving Strategies
Pytree to Dict Transformation
jax_sgmc.potential
Stochastic Potential
Full Potential
jax_sgmc.scheduler
Base Scheduler
Specific Schedulers
jax_sgmc.solver
MCMC
Solvers
Solver States
jax_sgmc.util
jax_sgmc.util.Array
jax_sgmc.util.tree_multiply
jax_sgmc.util.tree_add
jax_sgmc.util.tree_scale
jax_sgmc.util.list_vmap
jax_sgmc.util.list_pmap
jax_sgmc.util.pytree_list_to_leaves
jax_sgmc.util.pytree_leaves_to_list
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds