GitHub repository: mancusolab/traceax
Primary language: Python
Description: Stochastic trace estimation using JAX