Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | Q | R | S | T | U | V | X | Z _ __abs__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __add__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __array__() (quantax.utils.DataTracer method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __call__() (quantax.nn.RawInputLayer method) (quantax.nn.RefModel method) (quantax.state.DenseState method) (quantax.state.State method) (quantax.state.Variational method) __getitem__() (quantax.state.DenseState method) (quantax.state.State method) (quantax.utils.DataTracer method) __iadd__() (quantax.operator.Operator method) __imul__() (quantax.operator.Operator method) __init__() (quantax.model.GeneralDet method) (quantax.model.GeneralPf method) (quantax.model.MultiDet method) (quantax.model.MultiPf method) (quantax.model.ResConv method) (quantax.model.RestrictedDet method) (quantax.model.SingletPair method) (quantax.model.UnrestrictedDet method) (quantax.nn.ConvSymmetrize method) (quantax.nn.ReshapeConv method) (quantax.nn.Sequential method) (quantax.operator.Operator method) (quantax.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.optimizer.TimeEvol method) (quantax.sampler.ExactSampler method) (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.MixSampler method) (quantax.sampler.ParticleHop method) (quantax.sampler.RandomSampler method) (quantax.sampler.Sampler method) (quantax.sampler.Samples method) (quantax.sampler.SiteExchange method) (quantax.sampler.SiteFlip method) (quantax.sampler.SpinExchange method) (quantax.sites.Lattice method) (quantax.sites.Sites method) (quantax.state.DenseState method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.State method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) (quantax.symmetry.Symmetry method) (quantax.symmetry.Translation method) __isub__() (quantax.operator.Operator method) __itruediv__() (quantax.operator.Operator method) __jax_array__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __matmul__() (quantax.operator.Operator method) (quantax.state.State method) (quantax.symmetry.Symmetry method) __mul__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __neg__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __pow__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __radd__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __rmatmul__() (quantax.operator.Operator method) __rmul__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __rsub__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __rtruediv__() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __sub__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) __truediv__() (quantax.operator.Operator method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) A AdamSR (class in quantax.optimizer) annihilate() (in module quantax.operator) annihilate_d() (in module quantax.operator) annihilate_u() (in module quantax.operator) append() (quantax.utils.DataTracer method) apply_glorot_normal() (in module quantax.nn) 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) astype() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) auto_pinv_eig() (in module quantax.optimizer) B backward_chunk (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) basis (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) (quantax.symmetry.Symmetry property) (quantax.symmetry.Translation property) basis_make() (quantax.symmetry.Symmetry method) (quantax.symmetry.Translation 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.nn) character (quantax.symmetry.Symmetry property) (quantax.symmetry.Translation property) choose() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) chunk_map() (in module quantax.utils) chunk_shard_vmap() (in module quantax.utils) combine() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) compress() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) compute_sign() (in module quantax.nn) conj() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) ConvSymmetrize (class in quantax.nn) coord (quantax.sites.Lattice property) (quantax.sites.Sites property) copy() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) 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) diagonal() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) 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.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) E energy (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) ER (class in quantax.optimizer) ExactSampler (class in quantax.sampler) exp_by_log() (in module quantax.nn) exp_by_scale() (in module quantax.nn) expectation() (quantax.operator.Operator method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) expression (quantax.operator.Operator property) F F_full (quantax.model.GeneralPf property) (quantax.model.SingletPair property) fermion_idx() (in module quantax.nn) filter_replicate() (in module quantax.utils) filter_tree_map() (in module quantax.utils) flatten() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) Flip() (in module quantax.symmetry) forward_chunk (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) from_value() (quantax.utils.LogArray static method) (quantax.utils.ScaleArray static method) G GeneralDet (class in quantax.model) GeneralDetState (class in quantax.state) GeneralPf (class in quantax.model) GeneralPfState (class in quantax.state) get_default_dtype() (in module quantax) get_distribute_sharding() (in module quantax.utils) get_Ebar() (quantax.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.optimizer.TimeEvol method) get_lattice() (in module quantax) get_loss_fn() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) get_neighbor() (quantax.sites.Lattice method) (quantax.sites.Sites method) get_Obar() (quantax.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.optimizer.TimeEvol method) get_params_flatten() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) get_params_unflatten() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (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_SF() (quantax.optimizer.TimeEvol method) get_sites() (in module quantax) get_step() (quantax.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.optimizer.TimeEvol method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) get_subkeys() (in module quantax) get_sublattice_coord() (quantax.symmetry.Translation method) get_symm_spins() (quantax.symmetry.Symmetry method) (quantax.symmetry.Translation method) global_to_local() (in module quantax.utils) Grid (class in quantax.sites) H H (quantax.operator.Operator property) hamiltonian (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) Heisenberg() (in module quantax.operator) holomorphic (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.QNGD property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) Hubbard() (in module quantax.operator) I Identity() (in module quantax.symmetry) imag (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) imag_time (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.QNGD property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) index_from_xyz (quantax.sites.Lattice property) init_internal() (quantax.model.GeneralDet method) (quantax.model.GeneralPf method) (quantax.model.SingletPair method) (quantax.nn.RefModel method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) ints_to_array() (in module quantax.utils) is_default_cpl() (in module quantax) is_fermion (quantax.sites.Lattice property) (quantax.sites.Sites property) is_paired() (quantax.state.GeneralDetState class method) (quantax.state.GeneralPfState class method) (quantax.state.MeanFieldFermionState class method) (quantax.state.MultiDetState class method) (quantax.state.MultiPfState class method) (quantax.state.RestrictedDetState class method) (quantax.state.SingletPairState class method) (quantax.state.UnrestrictedDetState class method) is_sharded_array() (in module quantax.utils) is_spinful (quantax.sites.Lattice property) (quantax.sites.Sites property) Ising() (in module quantax.operator) J jacobian() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (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) LogArray (class in quantax.utils) M MARCH (class in quantax.optimizer) marshall_sign() (in module quantax.nn) mean() (quantax.utils.DataTracer method) (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) MeanFieldFermionState (class in quantax.state) Metropolis (class in quantax.sampler) MixSampler (class in quantax.sampler) model (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) mT (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) MultiDet (class in quantax.model) MultiDetState (class in quantax.state) MultiPf (class in quantax.model) MultiPfState (class in quantax.state) N nbytes (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) ncells (quantax.sites.Lattice property) ndim (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) neel() (in module quantax.utils) neel120_phase() (in module quantax.nn) nflips (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.MixSampler property) (quantax.sampler.ParticleHop property) (quantax.sampler.SiteExchange property) (quantax.sampler.SiteFlip property) (quantax.sampler.SpinExchange property) Nfmodes (quantax.sites.Lattice property) (quantax.sites.Sites property) Nmodes (quantax.sites.Lattice property) (quantax.sites.Sites property) norm() (quantax.state.DenseState method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.State method) (quantax.state.UnrestrictedDetState method) normalize() (quantax.model.MultiDet method) (quantax.state.DenseState method) (quantax.utils.ScaleArray method) normalize_() (quantax.state.DenseState method) nparams (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) Nparticles (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) nsamples (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.MixSampler property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) (quantax.sampler.SiteExchange property) (quantax.sampler.SiteFlip property) (quantax.sampler.SpinExchange property) Nsites (quantax.sites.Lattice property) (quantax.sites.Sites property) (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) nsymm (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) (quantax.symmetry.Symmetry property) (quantax.symmetry.Translation 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) orbitals() (quantax.sites.Lattice 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.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) permute_sign() (in module quantax.nn) plot() (quantax.sites.Lattice method) (quantax.sites.Sites method) (quantax.utils.DataTracer method) prod() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) prod_by_log() (in module quantax.nn) propose() (quantax.sampler.Metropolis method) (quantax.sampler.MixSampler method) psi (quantax.state.DenseState property) Pyrochlore (class in quantax.sites) Q QNGD (class in quantax.optimizer) R rand_states() (in module quantax.utils) RandomSampler (class in quantax.sampler) ravel() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) RawInputLayer (class in quantax.nn) RBM_Conv() (in module quantax.model) RBM_Dense() (in module quantax.model) real (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) reciprocal_vectors (quantax.sites.Lattice property) ref_chunk (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) ref_forward() (quantax.model.GeneralDet method) (quantax.model.GeneralPf method) (quantax.model.SingletPair method) (quantax.nn.RefModel method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) ref_forward_with_updates() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) RefModel (class in quantax.nn) repeat() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) ResConv (class in quantax.model) reset() (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.MixSampler method) (quantax.sampler.ParticleHop method) (quantax.sampler.SiteExchange method) (quantax.sampler.SiteFlip method) (quantax.sampler.SpinExchange method) reshape() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) ReshapeConv (class in quantax.nn) RestrictedDet (class in quantax.model) RestrictedDetState (class in quantax.state) reweight (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.MixSampler property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) (quantax.sampler.SiteExchange property) (quantax.sampler.SiteFlip property) (quantax.sampler.SpinExchange property) rho_from_model (quantax.state.GeneralDetState attribute) (quantax.state.GeneralPfState attribute) (quantax.state.MeanFieldFermionState attribute) (quantax.state.MultiDetState attribute) (quantax.state.MultiPfState attribute) (quantax.state.RestrictedDetState attribute) (quantax.state.SingletPairState attribute) (quantax.state.UnrestrictedDetState attribute) 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.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) (quantax.optimizer.TimeEvol method) (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) (quantax.utils.DataTracer method) save_time() (quantax.utils.DataTracer method) ScaleArray (class in quantax.utils) Sequential (class in quantax.nn) set_default_dtype() (in module quantax) set_random_seed() (in module quantax) shape (quantax.sites.Lattice property) (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) sharding (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) 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) SingletPair (class in quantax.model) SingletPairState (class in quantax.state) sinhp1_by_scale() (in module quantax.nn) site_offsets (quantax.sites.Lattice property) SiteExchange (class in quantax.sampler) SiteFlip (class in quantax.sampler) Sites (class in quantax.sites) size (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) solve() (quantax.optimizer.AdamSR method) (quantax.optimizer.ER method) (quantax.optimizer.MARCH method) (quantax.optimizer.QNGD method) (quantax.optimizer.SPRING method) (quantax.optimizer.SR method) SpinExchange (class in quantax.sampler) SpinInverse() (in module quantax.symmetry) SPRING (class in quantax.optimizer) Square() (in module quantax.sites) squeeze() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) Sqz_factor() (in module quantax.utils) SR (class in quantax.optimizer) State (class in quantax.state) state (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.QNGD property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) (quantax.sampler.ExactSampler property) (quantax.sampler.LocalFlip property) (quantax.sampler.Metropolis property) (quantax.sampler.MixSampler property) (quantax.sampler.ParticleHop property) (quantax.sampler.RandomSampler property) (quantax.sampler.Sampler property) (quantax.sampler.SiteExchange property) (quantax.sampler.SiteFlip property) (quantax.sampler.SpinExchange property) stripe() (in module quantax.utils) stripe_sign() (in module quantax.nn) sum() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) swapaxes() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) sweep() (quantax.sampler.ExactSampler method) (quantax.sampler.LocalFlip method) (quantax.sampler.Metropolis method) (quantax.sampler.MixSampler method) (quantax.sampler.ParticleHop method) (quantax.sampler.RandomSampler method) (quantax.sampler.Sampler method) (quantax.sampler.SiteExchange method) (quantax.sampler.SiteFlip method) (quantax.sampler.SpinExchange method) symm (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) symmetrize() (quantax.symmetry.Symmetry method) (quantax.symmetry.Translation method) Symmetry (class in quantax.symmetry) T T (quantax.utils.LogArray property) (quantax.utils.ScaleArray property) take() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) time (quantax.utils.DataTracer property) TimeEvol (class in quantax.optimizer) tJ() (in module quantax.operator) to_distribute_array() (in module quantax.utils) to_flax_model() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) 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.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.State method) (quantax.state.UnrestrictedDetState method) Translation (class in quantax.symmetry) TransND() (in module quantax.symmetry) transpose() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) 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) TriangularB (class in quantax.sites) tV() (in module quantax.operator) U U_full (quantax.model.GeneralDet property) (quantax.model.MultiDet property) (quantax.model.RestrictedDet property) (quantax.model.UnrestrictedDet property) uncertainty() (quantax.utils.DataTracer method) UnrestrictedDet (class in quantax.model) UnrestrictedDetState (class in quantax.state) update() (quantax.state.GeneralDetState method) (quantax.state.GeneralPfState method) (quantax.state.MeanFieldFermionState method) (quantax.state.MultiDetState method) (quantax.state.MultiPfState method) (quantax.state.RestrictedDetState method) (quantax.state.SingletPairState method) (quantax.state.UnrestrictedDetState method) (quantax.state.Variational method) use_ref (quantax.state.DenseState property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.State property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) V value() (quantax.utils.LogArray method) (quantax.utils.ScaleArray method) VarE (quantax.optimizer.AdamSR property) (quantax.optimizer.MARCH property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) Variational (class in quantax.state) vectors (quantax.symmetry.Translation property) VS_TYPE (class in quantax.state) vs_type (quantax.optimizer.AdamSR property) (quantax.optimizer.ER property) (quantax.optimizer.MARCH property) (quantax.optimizer.QNGD property) (quantax.optimizer.SPRING property) (quantax.optimizer.SR property) (quantax.optimizer.TimeEvol property) (quantax.state.GeneralDetState property) (quantax.state.GeneralPfState property) (quantax.state.MeanFieldFermionState property) (quantax.state.MultiDetState property) (quantax.state.MultiPfState property) (quantax.state.RestrictedDetState property) (quantax.state.SingletPairState property) (quantax.state.UnrestrictedDetState property) (quantax.state.Variational property) X xyz_from_index (quantax.sites.Lattice property) Z Z2_inversion (quantax.symmetry.Symmetry property) (quantax.symmetry.Translation property)