jonmorton/gateloop_jax

Implementation of gateloop transformer.

Python