awf/functional-transformer
A pure-functional implementation of a machine learning transformer model in Python/JAX
PythonMIT
Issues
- 1
Implementation of Attention is wrong.
#8 opened by AlexanderMath - 0
- 0
Add non-learned position encodings
#5 opened by awf - 3
Initialization of positional encodings?
#1 opened by btanner - 0