I am trying to understand the Google JAX framework, which is a collection of tools to speed up various forms of (matrix) computations.
In all the examples (e.g. official readthedocs.io) that I come across this is illustrated with comparing "naive" computations with JAX computations, using %timeit
. I kept running into errors until I discovered that this is a Jupyter/iPython magic function.
I think Jupyter/iPython is an amazing tool, especially for communicating something (a paper, etc.). But for tinkering I prefer a tool like Spyder or even Thonny, which makes it easier to keep track of what all the variables and functions in memory are.
What is the equivalent form of %timeit
that I can use to replace %timeit
in regular Python (from the timeit
package I assume)?