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.
bioRxiv · Jul 14, 2025 · PreprintBioinformatics