Lattice Field Theory Simulations meets JAX

JaxLatt is a high-performance lattice field theory simulation library built on JAX, designed for researchers in cosmology and high-energy physics. It provides efficient implementations of scalar and gauge field dynamics on discrete lattices, leveraging JAX's automatic differentiation and GPU acceleration capabilities.