GitHub repository: HPCForge/ParallelViT
Primary language: Jupyter Notebook
Description: FSDP and Pipeline parallelism in vit