JAX Pi

Posted on February 17, 2023

I've been using JAX for all sorts of things recently, but I hadn't yet implemented CBA's traditional HPC benchmark: calculting pi with a series expansion. Now I have: jaxpi.py.