Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z _ __add__() (quantax.symmetry.Symmetry method) __call__() (quantax.state.DenseState method) (quantax.state.State method) (quantax.state.Variational method) (quantax.symmetry.Symmetry method) __getitem__() (quantax.state.DenseState method) (quantax.state.State method) __init__() (quantax.nn.ConvSymmetrize method) (quantax.nn.Exp method) (quantax.nn.NoGradLayer method) (quantax.nn.Prod method) (quantax.nn.ReshapeConv method) (quantax.nn.Scale method) (quantax.nn.ScaleFn method) (quantax.nn.Sequential method) (quantax.nn.SinhShift method) (quantax.nn.Theta0Layer method) (quantax.operator.Operator method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.MinSR_Structured method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.sampler.ExactSampler method) (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.MixSampler method) (quantax.sampler.NeighborExchange method) (quantax.sampler.ParticleHop method) (quantax.sampler.RandomSampler method) (quantax.sampler.Sampler method) (quantax.sampler.Samples method) (quantax.sampler.SiteExchange method) (quantax.sites.Lattice method) (quantax.sites.Sites method) (quantax.state.DenseState method) (quantax.state.State method) (quantax.state.Variational method) (quantax.symmetry.Symmetry method) (quantax.utils.DataTracer method) __matmul__() (quantax.operator.Operator method) (quantax.state.State method) __rmatmul__() (quantax.operator.Operator method) A annihilate() (in module quantax.operator) annihilate_d() (in module quantax.operator) annihilate_u() (in module quantax.operator) append() (quantax.utils.DataTracer method) apply_he_normal() (in module quantax.nn) apply_lecun_normal() (in module quantax.nn) apply_updates() (in module quantax.utils) array_extend() (in module quantax.utils) array_set() (in module quantax.utils) array_to_ints() (in module quantax.utils) auto_pinv_eig() (in module quantax.optimizer) auto_shift_eig() (in module quantax.optimizer) B backward_chunk (quantax.state.Variational property) basis (quantax.state.DenseState property) (quantax.state.State property) (quantax.symmetry.Symmetry property) basis_make() (quantax.symmetry.Symmetry method) basis_vectors (quantax.sites.Lattice property) boundary (quantax.sites.Lattice property) C C4v() (in module quantax.symmetry) Chain() (in module quantax.sites) changed_inds() (in module quantax.utils) chunk_map() (in module quantax.utils) chunk_shard_vmap() (in module quantax.utils) combine() (quantax.state.Variational method) ConvSymmetrize (class in quantax.nn) coord (quantax.sites.Lattice property) (quantax.sites.Sites property) create() (in module quantax.operator) create_d() (in module quantax.operator) create_u() (in module quantax.operator) Cube() (in module quantax.sites) D D6() (in module quantax.symmetry) data (quantax.utils.DataTracer property) DataTracer (class in quantax.utils) DenseState (class in quantax.state) diagonalize() (quantax.operator.Operator method) dist (quantax.sites.Lattice property) (quantax.sites.Sites property) double_occ (quantax.sites.Lattice property) (quantax.sites.Sites property) dtype (quantax.state.Variational property) E eigval (quantax.symmetry.Symmetry property) ER (class in quantax.optimizer) ExactSampler (class in quantax.sampler) Exp (class in quantax.nn) expectation() (quantax.operator.Operator method) expression (quantax.operator.Operator property) F fermion_idx() (in module quantax.utils) filter_grad() (in module quantax.nn) filter_replicate() (in module quantax.utils) filter_tree_map() (in module quantax.utils) filter_vjp() (in module quantax.nn) Flip() (in module quantax.symmetry) forward_chunk (quantax.state.Variational property) G get_default_dtype() (in module quantax) get_Ebar() (quantax.optimizer.QNGD method) get_global_sharding() (in module quantax.utils) get_lattice() (in module quantax) get_neighbor() (quantax.sites.Lattice method) (quantax.sites.Sites method) get_Obar() (quantax.optimizer.QNGD method) get_params_flatten() (quantax.state.Variational method) get_params_unflatten() (quantax.state.Variational method) get_quspin_op() (quantax.operator.Operator method) get_real_dtype() (in module quantax) get_replicate_sharding() (in module quantax.utils) get_sites() (in module quantax) get_step() (quantax.optimizer.QNGD method) get_subkeys() (in module quantax) get_symm_spins() (quantax.symmetry.Symmetry method) global_to_local() (in module quantax.utils) Grid (class in quantax.sites) H H (quantax.operator.Operator property) Heisenberg() (in module quantax.operator) holomorphic (quantax.optimizer.QNGD property) (quantax.state.Variational property) Hubbard() (in module quantax.operator) I Identity() (in module quantax.symmetry) imag_time (quantax.optimizer.QNGD property) index_from_xyz (quantax.sites.Lattice property) init_internal() (quantax.state.Variational method) ints_to_array() (in module quantax.utils) is_balanced (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.NeighborExchange property) (quantax.sampler.ParticleHop property) is_fermion (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.symmetry.Symmetry property) is_sharded_array() (in module quantax.utils) is_spinful (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.symmetry.Symmetry property) Ising() (in module quantax.operator) J jacobian() (quantax.state.Variational method) jax_op_list (quantax.operator.Operator property) L Lattice (class in quantax.sites) LinearTransform() (in module quantax.symmetry) local_to_global() (in module quantax.utils) local_to_replicate() (in module quantax.utils) LocalFlip (class in quantax.sampler) M MARCH (class in quantax.optimizer) mean() (quantax.utils.DataTracer method) Metropolis (class in quantax.sampler) MinSR_Structured (class in quantax.optimizer) MixSampler (class in quantax.sampler) model (quantax.state.Variational property) N N (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.state.DenseState property) (quantax.state.State property) ncells (quantax.sites.Lattice property) ndim (quantax.sites.Lattice property) (quantax.sites.Sites property) neel() (in module quantax.utils) NeighborExchange (class in quantax.sampler) nflips (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.NeighborExchange property) (quantax.sampler.ParticleHop property) NoGradLayer (class in quantax.nn) norm() (quantax.state.DenseState method) (quantax.state.State method) normalize() (quantax.state.DenseState method) nparams (quantax.state.Variational property) Nparticle (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.state.DenseState property) (quantax.state.State property) nsamples (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.NeighborExchange property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) nstates (quantax.sites.Lattice property) (quantax.sites.Sites property) nsymm (quantax.state.DenseState property) (quantax.state.State property) (quantax.symmetry.Symmetry property) Ntotal (quantax.sites.Lattice property) (quantax.sites.Sites property) number() (in module quantax.operator) number_d() (in module quantax.operator) number_u() (in module quantax.operator) O Oloc() (quantax.operator.Operator method) op_list (quantax.operator.Operator property) Operator (class in quantax.operator) overlap() (quantax.state.DenseState method) (quantax.state.State method) P pair_cpl() (in module quantax.nn) PARTICLE_TYPE() (in module quantax) ParticleHole() (in module quantax.symmetry) ParticleHop (class in quantax.sampler) partition() (quantax.state.Variational method) permute_sign() (in module quantax.utils) plot() (quantax.sites.Lattice method) (quantax.sites.Sites method) (quantax.utils.DataTracer method) Prod (class in quantax.nn) Pyrochlore (class in quantax.sites) Q QNGD (class in quantax.optimizer) R rand_states() (in module quantax.utils) RandomSampler (class in quantax.sampler) RBM_Conv() (in module quantax.model) RBM_Dense() (in module quantax.model) ref_chunk (quantax.state.Variational property) ref_forward() (quantax.state.Variational method) ref_forward_with_updates() (quantax.state.Variational method) rescale() (quantax.nn.Exp method) (quantax.nn.Prod method) (quantax.nn.Sequential method) (quantax.nn.SinhShift method) (quantax.nn.Theta0Layer method) (quantax.state.Variational method) reset() (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.NeighborExchange method) (quantax.sampler.ParticleHop method) ReshapeConv (class in quantax.nn) ResProd() (in module quantax.model) ResSum() (in module quantax.model) reweight (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.NeighborExchange property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) Rotation() (in module quantax.symmetry) S S_m() (in module quantax.operator) S_p() (in module quantax.operator) S_x() (in module quantax.operator) S_y() (in module quantax.operator) S_z() (in module quantax.operator) Sampler (class in quantax.sampler) Samples (class in quantax.sampler) save() (quantax.state.Variational method) (quantax.utils.DataTracer method) save_time() (quantax.utils.DataTracer method) Scale (class in quantax.nn) ScaleFn (class in quantax.nn) SchmittNet() (in module quantax.model) Sequential (class in quantax.nn) set_default_dtype() (in module quantax) set_random_seed() (in module quantax) shape (quantax.sites.Lattice property) sharded_segment_sum() (in module quantax.utils) shmap() (in module quantax.utils) sigma_m() (in module quantax.operator) sigma_p() (in module quantax.operator) sigma_x() (in module quantax.operator) sigma_y() (in module quantax.operator) sigma_z() (in module quantax.operator) sign (quantax.sites.Lattice property) (quantax.sites.Sites property) SingleConv() (in module quantax.model) SingleDense() (in module quantax.model) SinhShift (class in quantax.nn) site_offsets (quantax.sites.Lattice property) SiteExchange (class in quantax.sampler) Sites (class in quantax.sites) solve() (quantax.optimizer.QNGD method) SpinInverse() (in module quantax.symmetry) SPRING (class in quantax.optimizer) Square() (in module quantax.sites) Sqz_factor() (in module quantax.utils) SR (class in quantax.optimizer) State (class in quantax.state) state (quantax.optimizer.QNGD property) (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.NeighborExchange property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) stripe() (in module quantax.utils) sweep() (quantax.sampler.ExactSampler method) (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.NeighborExchange method) (quantax.sampler.ParticleHop method) (quantax.sampler.RandomSampler method) (quantax.sampler.Sampler method) symm (quantax.state.DenseState property) (quantax.state.State property) (quantax.state.Variational property) symmetrize() (quantax.symmetry.Symmetry method) Symmetry (class in quantax.symmetry) T Theta0Layer (class in quantax.nn) time (quantax.utils.DataTracer property) tJ() (in module quantax.operator) to_flax_model() (quantax.state.Variational method) to_global_array() (in module quantax.utils) to_replicate_array() (in module quantax.utils) to_replicate_numpy() (in module quantax.utils) todense() (quantax.operator.Operator method) (quantax.state.DenseState method) (quantax.state.State method) Trans1D() (in module quantax.symmetry) Trans2D() (in module quantax.symmetry) Trans3D() (in module quantax.symmetry) Translation() (in module quantax.symmetry) TransND() (in module quantax.symmetry) tree_combine_cpl() (in module quantax.utils) tree_fully_flatten() (in module quantax.utils) tree_split_cpl() (in module quantax.utils) Triangular (class in quantax.sites) Triangular_ResSum() (in module quantax.model) TriangularB (class in quantax.sites) tV() (in module quantax.operator) U uncertainty() (quantax.utils.DataTracer method) update() (quantax.state.Variational method) V Variational (class in quantax.state) VS_TYPE (class in quantax.state) vs_type (quantax.optimizer.QNGD property) (quantax.state.Variational property) W wave_function (quantax.state.DenseState property) X xyz_from_index (quantax.sites.Lattice property) Z Z2_inversion (quantax.symmetry.Symmetry property)