GalacticDynamics / unxt

Unitful Quantities in JAX
https://unxt.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
11 stars 3 forks source link

Add benchmarking? #233

Open adrn opened 1 day ago

adrn commented 1 day ago

It might be good to have benchmarking comparisons for all unxt primitives and overrides against pure-numpy and pure-JAX to check for unexpected overheads in jitted and graded functions that contain units. But after doing some search, it seems like there are no well maintained benchmarking packages for modern Python!

nstarman commented 1 day ago

I know people use CodSpeed, which has the same API as pytest-benchmark. It looks really good. I've just had some trouble getting it to work with a Jax library on GH's VMs. When I run it locally it works.