/functional-transformer

A pure-functional implementation of a machine learning transformer model in Python/JAX

Primary LanguagePythonMIT LicenseMIT

Watchers

No one’s watching this repository yet.