The notebooks folder contains Jupyter notebooks, used to explain transformers using Jax and Flax. Author: David Cardozo.