Numpy-only re-implementation of https://arxiv.org/pdf/2303.18223 using mostly / or inspired by code from https://github.com/pageman/sutskever-30-implementations - View it on GitHub