Port of Mistral 7B in Keras and JAX - View it on GitHub