10.1101/2025.07.14.662216

traceax: a JAX-based framework for stochastic trace estimation

2025-07-18