GitHub repository: zzk1228/perceiver-io
Description: A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training