Technical requirements
In addition to the scientific Python packages NumPy and SciPy, we also need the SymPy, JAX, and diffrax
packages. These can be installed using your favorite package manager, such as pip
:
python3.10 -m pip install sympy jaxlib jax sympy diffrax
There are different options for the way you install JAX. Please see the official documentation for more details: https://github.com/google/jax#installation.
The code for this chapter can be found in the Chapter 03
folder of the GitHub repository at https://github.com/PacktPublishing/Applying-Math-with-Python-2nd-Edition/tree/main/Chapter%2003.