Pytorch Lightning code I tested out on the ASL competition on Kaggle: https://www.kaggle.com/competitions/asl-signs/overview.
Includes Transformer, GRU and Linear architecture, as well as a custom GRU layer (as the original one could not be converted to TF Lite).