Index _ | A | B | C | D | E | F | G | H | I | J | K | 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.MinSR method) (quantax.optimizer.QNGD method) (quantax.optimizer.TDVP method) (quantax.optimizer.TDVP_exact 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) 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) 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) det (in module quantax.utils) det_update_gen() (in module quantax.utils) det_update_rows() (in module quantax.utils) 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) energy (quantax.optimizer.MinSR property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact property) ExactSampler (class in quantax.sampler) Exp (class in quantax.nn) expectation() (quantax.operator.Operator method) expression (quantax.operator.Operator property) F 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.MinSR method) (quantax.optimizer.QNGD method) (quantax.optimizer.TDVP method) (quantax.optimizer.TDVP_exact 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.MinSR method) (quantax.optimizer.QNGD method) (quantax.optimizer.TDVP method) (quantax.optimizer.TDVP_exact 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.MinSR method) (quantax.optimizer.QNGD method) (quantax.optimizer.TDVP method) (quantax.optimizer.TDVP_exact method) get_subkeys() (in module quantax) get_symm_spins() (quantax.symmetry.Symmetry method) get_Tmat() (quantax.optimizer.MinSR method) global_to_local() (in module quantax.utils) Grid (class in quantax.sites) H H (quantax.operator.Operator property) hamiltonian (quantax.optimizer.MinSR property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact property) Heisenberg() (in module quantax.operator) holomorphic (quantax.optimizer.MinSR property) (quantax.optimizer.QNGD property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact property) (quantax.state.Variational property) Hubbard() (in module quantax.operator) I Identity() (in module quantax.symmetry) imag_time (quantax.optimizer.MinSR property) (quantax.optimizer.QNGD property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact property) index_from_xyz (quantax.sites.Lattice property) 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) K kazcmarz_mu (quantax.optimizer.MinSR property) (quantax.optimizer.QNGD property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact 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 mean() (quantax.utils.DataTracer method) Metropolis (class in quantax.sampler) MinSR (class in quantax.optimizer) minsr_pinv_eig() (in module 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 Ohvp() (quantax.optimizer.MinSR method) 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) pfa_eye() (in module quantax.utils) pfa_update() (in module quantax.utils) pfaffian (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) 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.MinSR method) (quantax.optimizer.QNGD method) (quantax.optimizer.TDVP method) (quantax.optimizer.TDVP_exact method) SpinInverse() (in module quantax.symmetry) Square() (in module quantax.sites) Sqz_factor() (in module quantax.utils) State (class in quantax.state) state (quantax.optimizer.MinSR property) (quantax.optimizer.QNGD property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact 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 TDVP (class in quantax.optimizer) TDVP_exact (class in quantax.optimizer) 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 VarE (quantax.optimizer.MinSR property) (quantax.optimizer.TDVP property) Variational (class in quantax.state) VS_TYPE (class in quantax.state) vs_type (quantax.optimizer.MinSR property) (quantax.optimizer.QNGD property) (quantax.optimizer.TDVP property) (quantax.optimizer.TDVP_exact 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)