1.2 JAX的安装与使用
JAX是一个最新的深度学习框架,但是当读者开始使用JAX的时候就会发现其非常简单。本节将带领读者安装JAX的开发环境,并演示第1个JAX的实战程序—MNIST数据集识别。
注意
目前,JAX开发环境必须搭建在Linux系统上,而对于大多数读者来说,安装和使用Linux是一个较为困难的操作,因此这里采用Windows 10或者更高版本自带的虚拟机形式安装JAX开发环境。不用担心,只需要安装一遍并跟随笔者手把手地配置好开发环境和工具,之后就可以像开发传统的Windows程序一样进行JAX程序的开发。