Tutorials

Training Models at Scale

I gave a tutorial on distributed training strategies for large-scale models.

Training Models at Scale Tutorials

Deep Learning Tutorials on Training Models at Scale in JAX with Flax

GDE: Machine Learning with JAX and Flax

Introduction talk for learning JAX and Flax as Deep Learning framework

Deep Learning Tutorials Translated to JAX with Flax

Deep Learning Tutorials in PyTorch with PyTorch Lightning and JAX with Flax

Official Collaboration with PyTorch Lightning

University of Amsterdam Deep Learning Tutorials now part of PyTorch Lightning