谷歌JAX深度学习从零开始学
上QQ阅读APP看书,第一时间看更新

1.2 JAX的安装与使用

JAX是一个最新的深度学习框架,但是当读者开始使用JAX的时候就会发现其非常简单。本节将带领读者安装JAX的开发环境,并演示第1个JAX的实战程序—MNIST数据集识别。

注意

目前,JAX开发环境必须搭建在Linux系统上,而对于大多数读者来说,安装和使用Linux是一个较为困难的操作,因此这里采用Windows 10或者更高版本自带的虚拟机形式安装JAX开发环境。不用担心,只需要安装一遍并跟随笔者手把手地配置好开发环境和工具,之后就可以像开发传统的Windows程序一样进行JAX程序的开发。