Fast optimization of IQP circuits with JAX - Recio-Armengol, Erik; Bowles, Joseph