【假一罚四】JAX可微分编程

本书以Google开发的JAX开源框架为载体，详细介绍了JAX在可微分编程领域的应用，具体包括自动微分的基本原理、数据结构，以及自动微分在实际场景中的应用，其涉及的领域包括但不限于算法优化、神经网络、工程建模、量子计算等。

本书分为3部分，总计10章外加5篇附录。第1部分介绍了可微分编程的基本原理，包括手动求导、数值微分、符号微分以及自动微分的前向模式和反向模式，在未调用任何库函数的情况下，从零开始构建起了符号微分及自动微分的数据结构。第2部分是对JAX库特性的介绍，包括JAX的基本语法、自动微分、即时编译和并行计算，并以此为切口，对深度学习、λ演算等领域进行了深入浅出的讲解。第3部分是自动微分在实际场景中的应用，除了算法优化、神经网络等极其经典的应用场景，这一部分还给出了自动微分在工程建模、量子计算等方面的理论及应用。

本书涵盖的范围虽广，但对其中每个领域的介绍都绝非浅尝辄止，无论材料的选取、内容的编排，还是论述的视角、观点的呈现，均不乏新颖之处。通过本书的学习，读者不但可以掌握JAX开源框架的用法，还可以获悉JAX在可微分编程领域的具体应用方法。本书适合在工作中会用到自动微分技术的工程技术人员、高校科研人员阅读，也适合对JAX感兴趣并希望能掌握其应用的AI技术人员阅读。