awf/functional-transformer

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

语言：Python · GitHub 星标：185 · Fork：8

在 GitHub 上查看原仓库