带你入门前沿深度学习框架JAX
1. 了解Linux基础操作 2. 了解Python3基础 3. 了解机器学习和深度学习基础
1、深度学习框架的发展(Pytorch、TF和JAX) 2、JAX基础矩阵计算 3、JAX梯度计算与参数优化 4、JAX深度学习模型搭建 5、JAX模型保存与加载