Skip to main content
Ctrl+K

Quantax 0.1.1 documentation

  • Installation
  • API
  • Installation
  • API

Section Navigation

  • global_defs
  • sites
    • quantax.sites.Sites
    • quantax.sites.Lattice
    • quantax.sites.Grid
    • quantax.sites.Chain
    • quantax.sites.Square
    • quantax.sites.Cube
    • quantax.sites.Pyrochlore
    • quantax.sites.Triangular
    • quantax.sites.TriangularB
  • symmetry
    • quantax.symmetry.Symmetry
    • quantax.symmetry.Identity
    • quantax.symmetry.SpinInverse
    • quantax.symmetry.ParticleHole
    • quantax.symmetry.Translation
    • quantax.symmetry.TransND
    • quantax.symmetry.Trans1D
    • quantax.symmetry.Trans2D
    • quantax.symmetry.Trans3D
    • quantax.symmetry.LinearTransform
    • quantax.symmetry.Flip
    • quantax.symmetry.Rotation
    • quantax.symmetry.C4v
    • quantax.symmetry.D6
  • operator
    • quantax.operator.Operator
    • quantax.operator.sigma_x
    • quantax.operator.sigma_y
    • quantax.operator.sigma_z
    • quantax.operator.sigma_p
    • quantax.operator.sigma_m
    • quantax.operator.S_x
    • quantax.operator.S_y
    • quantax.operator.S_z
    • quantax.operator.S_p
    • quantax.operator.S_m
    • quantax.operator.create
    • quantax.operator.create_u
    • quantax.operator.create_d
    • quantax.operator.annihilate
    • quantax.operator.annihilate_u
    • quantax.operator.annihilate_d
    • quantax.operator.number
    • quantax.operator.number_u
    • quantax.operator.number_d
    • quantax.operator.Heisenberg
    • quantax.operator.Ising
    • quantax.operator.Hubbard
    • quantax.operator.tJ
    • quantax.operator.tV
  • nn
    • quantax.nn.Sequential
    • quantax.nn.NoGradLayer
    • quantax.nn.filter_grad
    • quantax.nn.filter_vjp
    • quantax.nn.Scale
    • quantax.nn.ScaleFn
    • quantax.nn.Theta0Layer
    • quantax.nn.SinhShift
    • quantax.nn.Prod
    • quantax.nn.Exp
    • quantax.nn.pair_cpl
    • quantax.nn.apply_lecun_normal
    • quantax.nn.apply_he_normal
    • quantax.nn.ReshapeConv
    • quantax.nn.ConvSymmetrize
  • model
    • quantax.model.SingleDense
    • quantax.model.RBM_Dense
    • quantax.model.SingleConv
    • quantax.model.RBM_Conv
    • quantax.model.ResProd
    • quantax.model.SchmittNet
    • quantax.model.ResSum
    • quantax.model.Triangular_ResSum
  • state
    • quantax.state.State
    • quantax.state.DenseState
    • quantax.state.Variational
    • quantax.state.VS_TYPE
  • sampler
    • quantax.sampler.Sampler
    • quantax.sampler.Samples
    • quantax.sampler.ExactSampler
    • quantax.sampler.RandomSampler
    • quantax.sampler.Metropolis
    • quantax.sampler.LocalFlip
    • quantax.sampler.NeighborExchange
    • quantax.sampler.ParticleHop
    • quantax.sampler.SiteExchange
    • quantax.sampler.MixSampler
  • optimizer
    • quantax.optimizer.QNGD
    • quantax.optimizer.TDVP
    • quantax.optimizer.TDVP_exact
    • quantax.optimizer.MinSR
    • quantax.optimizer.auto_pinv_eig
    • quantax.optimizer.minsr_pinv_eig
  • utils
    • quantax.utils.DataTracer
    • quantax.utils.get_global_sharding
    • quantax.utils.get_replicate_sharding
    • quantax.utils.is_sharded_array
    • quantax.utils.to_global_array
    • quantax.utils.to_replicate_array
    • quantax.utils.global_to_local
    • quantax.utils.local_to_global
    • quantax.utils.local_to_replicate
    • quantax.utils.to_replicate_numpy
    • quantax.utils.array_extend
    • quantax.utils.array_set
    • quantax.utils.sharded_segment_sum
    • quantax.utils.ints_to_array
    • quantax.utils.array_to_ints
    • quantax.utils.neel
    • quantax.utils.stripe
    • quantax.utils.Sqz_factor
    • quantax.utils.rand_states
    • quantax.utils.tree_fully_flatten
    • quantax.utils.filter_replicate
    • quantax.utils.filter_tree_map
    • quantax.utils.tree_split_cpl
    • quantax.utils.tree_combine_cpl
    • quantax.utils.apply_updates
    • quantax.utils.chunk_shard_vmap
    • quantax.utils.chunk_map
    • quantax.utils.shmap
    • quantax.utils.det
    • quantax.utils.pfaffian
    • quantax.utils.det_update_rows
    • quantax.utils.det_update_gen
    • quantax.utils.pfa_eye
    • quantax.utils.pfa_update
  • API

API#

  • global_defs
    • set_default_dtype()
    • get_default_dtype()
    • get_real_dtype()
    • set_random_seed()
    • get_subkeys()
    • PARTICLE_TYPE()
    • get_sites()
    • get_lattice()
  • sites
    • Main classes
    • Common lattices
  • symmetry
    • Main class
    • Common symmetries
  • operator
    • Main class
    • Site operators
    • Hamiltonians
  • nn
    • Modules
    • Activation function
    • Initializers
    • NQS layers
  • model
    • Neural quantum states
  • state
    • Main classes
  • sampler
    • Main classes
    • Exact samplers
    • Metropolis samplers
  • optimizer
    • Quantum natural gradient descent
    • Solvers
  • utils
    • Data
    • Sharding
    • Array
    • Spins
    • Pytree
    • Function
    • Linear algebra

previous

Installation

next

global_defs

This Page

  • Show Source

© Copyright 2024, Ao Chen.

Created using Sphinx 8.2.3.

Built with the PyData Sphinx Theme 0.16.1.