BioinformaticsbioRxivPreprint
traceax: a JAX-based framework for stochastic trace estimation
Abdullah Al Nahid, Linda Serafin, Nicholas Mancuso
traceax provides a scalable framework for stochastic trace estimation, offering efficient and accurate trace computation with reduced memory and runtime compared to direct methods.
DOI 10.1101/2025.07.14.662216