Similar repositories to sifanw094/vanilla-transformer-jax: