note: in my bumbling i found this doc which gives a general intro to flax/jax/huggingface from google: https://github.com/huggingface/transformers/blob/master/examples/research_projects/jax-projects/README.md . i'm wondering if stuff like that doc is how jax reached me.