awf/functional-transformer
A pure-functional implementation of a machine learning transformer model in Python/JAX
Language: Python
Stars: 104 Issues: 1 Forks: 2
https://github.com/awf/functional-transformer
A pure-functional implementation of a machine learning transformer model in Python/JAX - GitHub - awf/functional-transformer: A pure-functional implementation of a machine learning transformer mode...
