More info can be found at https://github.com/jzhang38/LongMamba