JAX-based framework for Lattice Hamiltonian simulation - View it on GitHub