Temporary patched version of sequence layers w/ MLX support - View it on GitHub