简介:在课程中我们将使用最前沿的JAX框架,带你来感受体验最为真实的深度学习的算法开发过程。课程内容从算法原理到算法实战,从思路到代码,从框架到具体模型编码,助力你轻松入门前沿框架JAX。
第1章 JAX基础概览
与常见深度学习进行对比,并介绍了JAX的起源、特点和基础案例。从本章开始带领同学使用JAX解决基础的建模问题,并为之后的内容提供知识铺垫。
第2章 JAX极速入门
讲解了JAX的安装方法和使用流程,通过房价预测和图像分类带领同学们学会使用JAX解决实际的问题,包括损失函数计算、梯度计算和参数更新的细节。
第3章 JAX搭建服饰分类模型
在JAX基础知识点的基础上,深入JAX的实现过程。本章将通过JAX搭建深度学习模型,从数据集读取到模型搭建,讲解JAX模型搭建的细节流程。
第4章 JAX训练服饰分类模型
在服饰分类模型的基础上,讲解JAX损失函数、训练流程、模型保存和预测。最终带领同学完成深度学习模型的训练流程,完成实际模型的训练和调优。