



除了科学Python软件包NumPy和SciPy,我们还需要SymPy、JAX和diffrax包。可以使用你喜欢的软件包管理器(例如pip)安装这些软件包:
安装JAX的方式有多种选择,更多详细信息请参阅官方文档:https://github.com/google/jax#installation。
本章的代码可在GitHub代码库的“Chapter 03”文件夹中找到,网址为https://github.com/PacktPublishing/Applying-Math-with-Python-2nd-Edition/tree/main/Chapter%2003。