Full Changelog: v3.22.0dev2...v3.22.0dev3

Full Changelog: v3.22.0dev1...v3.22.0dev2

Full Changelog: v3.21.0...v3.22.0dev1

What's Changed

• Add JaxOperatorGetConnInJitError for better error handling by @PhilipVinc in #2170
• Simplify lattice group representation contructor. by @PhilipVinc in #2155
• Extend infidelity_sr to FullSumState by @Z-Denis in #2171
• Fix bug in transformed vstate by @louishrm in #2175
• support complex molecular orbitals by @inailuig in #2177
• Distribute mo_coeff from process 0 by @inailuig in #2178
• Track 'order' to fermion operator 2nd by @PhilipVinc in #2180
• Move PNC operators into nk._src to avoid import loops by @PhilipVinc in #2179
• Fix a bug in ParticleNumberAndSpinConservingFermioperator2nd where the ordering of spin sectors was reversed by @inailuig in #2184
• Fix a bug where we would sneak arguments with auto sharding into the shard_map by @inailuig in #2182
• Introduce flax/nnx wrappers to do operator@vstate and allow for nesting them by @PhilipVinc in #2186
• Update sphinx requirement from ~=8.2.3 to >=8.2.3,<9.1.0 by @dependabot[bot] in #2188
• Implement a DeterminantVariationalState that computes expectations/gradients exactly by @PhilipVinc in #2181

Full Changelog: v3.20.3...v3.21.0

• Fix a bug in the product between Ising @ other_operator being wrongly computed as other @ Ising. This also affected Heisenberg.
• Fix a bug where the ProductOperator was wrongly computing the product of two operators.
  Full Changelog: v3.20.4...v3.20.5

What's Changed

• Add JaxOperatorGetConnInJitError for better error handling by @PhilipVinc in #2170
• Fix Bug in nk.vqs.apply_operator

Full Changelog: v3.20.3...v3.20.4

Fix a minor issue with VMC_SR(on_the_fly=False) not sharding correctly.

Full Changelog: v3.20.2...v3.20.3

Fix a minor NNXWrapper bug

Full Changelog: v3.20.1...v3.20.2

Support jax 0.8

What's Changed

• Deprecate nk.jax.tree_ravel in favor of JAX's ravel_pytree (JAX 0.8 compatibility) by @PhilipVinc in #2164

Full Changelog: v3.20.0...v3.20.1