Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch