replace numerical derivatives with jax?

LatticeQCD / AnalysisToolbox

A set of Python tools for statistically analyzing data. This includes aspects of lattice QCD applications related to QCD phenomenology.

MIT License

22 stars 4 forks source link

replace numerical derivatives with jax? #58

Open clarkedavida opened 4 months ago

clarkedavida commented 4 months ago

right now we do our numerical derivatives with python for-loops so they are incredibly slow. i have wanted to replace them for a while, but only today learned of jax

https://github.com/google/jax

which i think is worth trying