econser/jaxformer

Learning project for jax. I'm building a tokenizer, word embedder, and transformer in jax & flax

Python