learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-12-11 10:57:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

link to jax transformer

Browse Source
This commit is contained in:
Varuna Jayasiri
2025-11-11 09:22:38 +00:00
parent c3d868baa1
commit 25e169843e
3 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/index.html
View File

@@ -79,7 +79,8 @@
<h3><strong><a href="https://nn.labml.ai/ja/">Japanese (translated)</a></strong></h3>
<h2>Paper Implementations</h2>
<h4>✨ <a href="transformers/index.html">Transformers</a></h4>
<ul><li><a href="transformers/mha.html">Multi-headed attention</a> </li>
<ul><li><a href="transformers/jax_transformer/index.html">JAX implementation</a> </li>
<li><a href="transformers/mha.html">Multi-headed attention</a> </li>
<li><a href="transformers/flash/index.html">Triton Flash Attention</a> </li>
<li><a href="transformers/models.html">Transformer building blocks</a> </li>
<li><a href="transformers/xl/index.html">Transformer XL</a> </li>

1
labml_nn/__init__.py
View File

@@ -24,6 +24,7 @@ implementations.
#### ✨ [Transformers](transformers/index.html)
* [JAX implementation](transformers/jax_transformer/index.html)
* [Multi-headed attention](transformers/mha.html)
* [Triton Flash Attention](transformers/flash/index.html)
* [Transformer building blocks](transformers/models.html)

1
readme.md
View File

@@ -20,6 +20,7 @@ implementations almost weekly.
#### ✨ [Transformers](https://nn.labml.ai/transformers/index.html)
* [JAX implementation](https://nn.labml.ai/transformers/jax_transformer/index.html)
* [Multi-headed attention](https://nn.labml.ai/transformers/mha.html)
* [Triton Flash Attention](https://nn.labml.ai/transformers/flash/index.html)
* [Transformer building blocks](https://nn.labml.ai/transformers/models.html)