[Take-2] Infrastructure for Jax-based operators, IsingJax (#1506)

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

replaces #1335 closes #1335

---------

Co-authored-by: Dian Wu <dian.wu@epfl.ch>
Co-authored-by: Clemens Giuliani <clemens@inailuig.it>