This folder contains code for the Flax implementation of the PEGASUS model, as well as the PEGASUS-X model that is adapted for long input summarization. The PEGASUS-X model is described in ...