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
Previous Next

© Copyright 2021, Multiscale Modeling of Fluid Materials, TU Munich. Revision 5f5fd46b.

Built with Sphinx using a theme provided by Read the Docs.