Self-attention mechanism implementd in Jax
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.