#JAX+Flax

这套深度学习教程有助于理解理论知识，涵盖优化、Transformer、图神经网络等主题。基于PyTorch和PyTorch Lightning框架，并提供JAX+Flax实现。教程支持本地运行、Google Colab和Snellius集群，多种方式供选择。每个教程包含详细的笔记本，实现理论与实践相结合。本课程与正式作业和考试相关，适合想深入了解深度学习及应用的学习者。