awf / functional-transformer

A pure-functional implementation of a machine learning transformer model in Python/JAX
175Updated 2 years ago

Related projects

Alternatives and complementary repositories for functional-transformer