Search results for jax statistical physics