/self-attention-jax

Self-attention mechanism implementd in Jax

Primary LanguagePythonMIT LicenseMIT

Watchers