本书作为一本可微分编程领域的专著,以Google开发的Jax开源库为载体,介绍了自动微分的基本原理,数据结构,以及自动微分在实际场景下的应用,其涉及的领域包括但不限于优化算法、神经网络、工程建模、量子计算等。本书分为3个部分,总计10章。第1部分介绍了可微分编程的基本原理,包括手动求导、数值微分、符号微分以及自动微分的前向和反向模式;且在未调用任何库函数的情况下,从零开始构建起了符号微分及自动微分的数据结构。第2部分是对Jax库特性的介绍,包括Jax的基本语法、自动微分、即时编译和并行计算,并以此为切口,