Sampling using tnco.app.circuit.Sampler.sample with the flag contraction_backend is set to 'jax'. Most likely, the issue is related to the internal contraction routines tnco.utils.tensor.tensordot and tnco.utils.tn.contraction, which would require multiple warmups for a JIT library as jax.