[Take-2] Infrastructure for Jax-based operators, IsingJax (!1506) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge jax_operators into master Jun 27, 2023

Created by: PhilipVinc

This builds on top of the code written originally by @wdphy16 and makes fully jax-compliant Ising operators as well as lays the groundwork for similar changes around...

replaces #1335 closes #1335

[Take-2] Infrastructure for Jax-based operators, IsingJax

Merge request reports