JAX, Flax & Transformers — State-of-the-Art ML Models | Open Awesome