GalacticDynamics / unxt

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

Add benchmarking? #233

Open adrn opened 1 month ago

adrn commented 1 month 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 month 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.

nstarman commented 3 weeks ago

See #265. It's partially working.