diff --git a/docs/activations/fta/experiment.html b/docs/activations/fta/experiment.html
index 1a92fe94..db9a8e39 100644
--- a/docs/activations/fta/experiment.html
+++ b/docs/activations/fta/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/activations/fta/index.html b/docs/activations/fta/index.html
index f0130207..b0c2fd45 100644
--- a/docs/activations/fta/index.html
+++ b/docs/activations/fta/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/activations/index.html b/docs/activations/index.html
index 2d7a5f46..4a04c3b1 100644
--- a/docs/activations/index.html
+++ b/docs/activations/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/activations/swish.html b/docs/activations/swish.html
index 9d55187c..d6f60067 100644
--- a/docs/activations/swish.html
+++ b/docs/activations/swish.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/index.html b/docs/adaptive_computation/index.html
index 5400aada..00b0b632 100644
--- a/docs/adaptive_computation/index.html
+++ b/docs/adaptive_computation/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/parity.html b/docs/adaptive_computation/parity.html
index 24220ecf..89807a1e 100644
--- a/docs/adaptive_computation/parity.html
+++ b/docs/adaptive_computation/parity.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/ponder_net/experiment.html b/docs/adaptive_computation/ponder_net/experiment.html
index 3308ec0f..024f770b 100644
--- a/docs/adaptive_computation/ponder_net/experiment.html
+++ b/docs/adaptive_computation/ponder_net/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/ponder_net/index.html b/docs/adaptive_computation/ponder_net/index.html
index c4a40840..16800830 100644
--- a/docs/adaptive_computation/ponder_net/index.html
+++ b/docs/adaptive_computation/ponder_net/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/ponder_net/readme.html b/docs/adaptive_computation/ponder_net/readme.html
index 938e20eb..7c7ae6a7 100644
--- a/docs/adaptive_computation/ponder_net/readme.html
+++ b/docs/adaptive_computation/ponder_net/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/adaptive_computation/readme.html b/docs/adaptive_computation/readme.html
index 6e5f8623..e9f27ea5 100644
--- a/docs/adaptive_computation/readme.html
+++ b/docs/adaptive_computation/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/capsule_networks/index.html b/docs/capsule_networks/index.html
index 267adcf1..b8ae6fe7 100644
--- a/docs/capsule_networks/index.html
+++ b/docs/capsule_networks/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/capsule_networks/mnist.html b/docs/capsule_networks/mnist.html
index eac5bb59..abd5d731 100644
--- a/docs/capsule_networks/mnist.html
+++ b/docs/capsule_networks/mnist.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/capsule_networks/readme.html b/docs/capsule_networks/readme.html
index 0f72e101..82f47d4d 100644
--- a/docs/capsule_networks/readme.html
+++ b/docs/capsule_networks/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/cfr/analytics.html b/docs/cfr/analytics.html
index 79ce2c28..f5873612 100644
--- a/docs/cfr/analytics.html
+++ b/docs/cfr/analytics.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/cfr/index.html b/docs/cfr/index.html
index e172c1fc..5148065b 100644
--- a/docs/cfr/index.html
+++ b/docs/cfr/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/cfr/infoset_saver.html b/docs/cfr/infoset_saver.html
index 2d8749e8..21abaa26 100644
--- a/docs/cfr/infoset_saver.html
+++ b/docs/cfr/infoset_saver.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/cfr/kuhn/index.html b/docs/cfr/kuhn/index.html
index be4fd6a6..167172fb 100644
--- a/docs/cfr/kuhn/index.html
+++ b/docs/cfr/kuhn/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/conv_mixer/experiment.html b/docs/conv_mixer/experiment.html
index 06f7fcfc..848e8106 100644
--- a/docs/conv_mixer/experiment.html
+++ b/docs/conv_mixer/experiment.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/conv_mixer/index.html b/docs/conv_mixer/index.html
index 687d1c28..c29a857b 100644
--- a/docs/conv_mixer/index.html
+++ b/docs/conv_mixer/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/conv_mixer/readme.html b/docs/conv_mixer/readme.html
index dd47601f..7eae80d4 100644
--- a/docs/conv_mixer/readme.html
+++ b/docs/conv_mixer/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/evaluate.html b/docs/diffusion/ddpm/evaluate.html
index 4629b3b8..3bfc08f2 100644
--- a/docs/diffusion/ddpm/evaluate.html
+++ b/docs/diffusion/ddpm/evaluate.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/experiment.html b/docs/diffusion/ddpm/experiment.html
index f0d75118..03aaea0d 100644
--- a/docs/diffusion/ddpm/experiment.html
+++ b/docs/diffusion/ddpm/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/index.html b/docs/diffusion/ddpm/index.html
index dd913e51..27f1f4a2 100644
--- a/docs/diffusion/ddpm/index.html
+++ b/docs/diffusion/ddpm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/readme.html b/docs/diffusion/ddpm/readme.html
index 0fdc16e3..68d82b59 100644
--- a/docs/diffusion/ddpm/readme.html
+++ b/docs/diffusion/ddpm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/unet.html b/docs/diffusion/ddpm/unet.html
index ac4186b1..fccc9977 100644
--- a/docs/diffusion/ddpm/unet.html
+++ b/docs/diffusion/ddpm/unet.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/ddpm/utils.html b/docs/diffusion/ddpm/utils.html
index f0d9d2d5..11515630 100644
--- a/docs/diffusion/ddpm/utils.html
+++ b/docs/diffusion/ddpm/utils.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/diffusion/index.html b/docs/diffusion/index.html
index c6850bad..a9e9ba1c 100644
--- a/docs/diffusion/index.html
+++ b/docs/diffusion/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/distillation/index.html b/docs/distillation/index.html
index ac24f4f1..4c23dbcc 100644
--- a/docs/distillation/index.html
+++ b/docs/distillation/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/distillation/large.html b/docs/distillation/large.html
index 99b0975b..ab6afc06 100644
--- a/docs/distillation/large.html
+++ b/docs/distillation/large.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/distillation/readme.html b/docs/distillation/readme.html
index 84edcd1a..74c52600 100644
--- a/docs/distillation/readme.html
+++ b/docs/distillation/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/distillation/small.html b/docs/distillation/small.html
index 68ab190b..40230b74 100644
--- a/docs/distillation/small.html
+++ b/docs/distillation/small.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/arithmetic_dataset.html b/docs/experiments/arithmetic_dataset.html
index 7501016b..38daa510 100644
--- a/docs/experiments/arithmetic_dataset.html
+++ b/docs/experiments/arithmetic_dataset.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/cifar10.html b/docs/experiments/cifar10.html
index 7b57934c..eeb619b3 100644
--- a/docs/experiments/cifar10.html
+++ b/docs/experiments/cifar10.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/index.html b/docs/experiments/index.html
index db2192cf..bf1253f4 100644
--- a/docs/experiments/index.html
+++ b/docs/experiments/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/mnist.html b/docs/experiments/mnist.html
index 815dbaa1..5a50610c 100644
--- a/docs/experiments/mnist.html
+++ b/docs/experiments/mnist.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/nlp_autoregression.html b/docs/experiments/nlp_autoregression.html
index 98c07602..3447d2e7 100644
--- a/docs/experiments/nlp_autoregression.html
+++ b/docs/experiments/nlp_autoregression.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/experiments/nlp_classification.html b/docs/experiments/nlp_classification.html
index 7cbe8bcd..dcbeed68 100644
--- a/docs/experiments/nlp_classification.html
+++ b/docs/experiments/nlp_classification.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/cycle_gan/index.html b/docs/gan/cycle_gan/index.html
index b0bdf5dd..7c4862a6 100644
--- a/docs/gan/cycle_gan/index.html
+++ b/docs/gan/cycle_gan/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/cycle_gan/readme.html b/docs/gan/cycle_gan/readme.html
index f704802d..2aee81a7 100644
--- a/docs/gan/cycle_gan/readme.html
+++ b/docs/gan/cycle_gan/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/dcgan/index.html b/docs/gan/dcgan/index.html
index 8f21ca90..05d44f46 100644
--- a/docs/gan/dcgan/index.html
+++ b/docs/gan/dcgan/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/dcgan/readme.html b/docs/gan/dcgan/readme.html
index 0e0aee2a..394b9448 100644
--- a/docs/gan/dcgan/readme.html
+++ b/docs/gan/dcgan/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/index.html b/docs/gan/index.html
index 0daee8e7..14bfcfbe 100644
--- a/docs/gan/index.html
+++ b/docs/gan/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/original/experiment.html b/docs/gan/original/experiment.html
index e4ebfef8..39b23b82 100644
--- a/docs/gan/original/experiment.html
+++ b/docs/gan/original/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/original/index.html b/docs/gan/original/index.html
index a4b2b15b..35aa6c8c 100644
--- a/docs/gan/original/index.html
+++ b/docs/gan/original/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/original/readme.html b/docs/gan/original/readme.html
index 33a6a632..6ca6cf47 100644
--- a/docs/gan/original/readme.html
+++ b/docs/gan/original/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/stylegan/experiment.html b/docs/gan/stylegan/experiment.html
index 93290ad3..1fa07ff1 100644
--- a/docs/gan/stylegan/experiment.html
+++ b/docs/gan/stylegan/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/stylegan/index.html b/docs/gan/stylegan/index.html
index 14f4d146..75649dbe 100644
--- a/docs/gan/stylegan/index.html
+++ b/docs/gan/stylegan/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/stylegan/readme.html b/docs/gan/stylegan/readme.html
index 9b2a946d..8a917138 100644
--- a/docs/gan/stylegan/readme.html
+++ b/docs/gan/stylegan/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/experiment.html b/docs/gan/wasserstein/experiment.html
index 929945dd..b71c631f 100644
--- a/docs/gan/wasserstein/experiment.html
+++ b/docs/gan/wasserstein/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/gradient_penalty/experiment.html b/docs/gan/wasserstein/gradient_penalty/experiment.html
index 197b86b3..a32cdd3e 100644
--- a/docs/gan/wasserstein/gradient_penalty/experiment.html
+++ b/docs/gan/wasserstein/gradient_penalty/experiment.html
@@ -53,8 +53,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/gradient_penalty/index.html b/docs/gan/wasserstein/gradient_penalty/index.html
index 184eb1a2..0d9cc9b3 100644
--- a/docs/gan/wasserstein/gradient_penalty/index.html
+++ b/docs/gan/wasserstein/gradient_penalty/index.html
@@ -56,8 +56,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/gradient_penalty/readme.html b/docs/gan/wasserstein/gradient_penalty/readme.html
index b1f6b190..d1b446a4 100644
--- a/docs/gan/wasserstein/gradient_penalty/readme.html
+++ b/docs/gan/wasserstein/gradient_penalty/readme.html
@@ -53,8 +53,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/index.html b/docs/gan/wasserstein/index.html
index e05f38e9..a7b9a1ae 100644
--- a/docs/gan/wasserstein/index.html
+++ b/docs/gan/wasserstein/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/gan/wasserstein/readme.html b/docs/gan/wasserstein/readme.html
index 9a9f5643..88b99582 100644
--- a/docs/gan/wasserstein/readme.html
+++ b/docs/gan/wasserstein/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gat/experiment.html b/docs/graphs/gat/experiment.html
index fa0047ff..f5fecf10 100644
--- a/docs/graphs/gat/experiment.html
+++ b/docs/graphs/gat/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gat/index.html b/docs/graphs/gat/index.html
index c9a7326b..42c023f3 100644
--- a/docs/graphs/gat/index.html
+++ b/docs/graphs/gat/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gat/readme.html b/docs/graphs/gat/readme.html
index 43910de7..c7ac59e1 100644
--- a/docs/graphs/gat/readme.html
+++ b/docs/graphs/gat/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gatv2/experiment.html b/docs/graphs/gatv2/experiment.html
index e7b810ae..2b51ef03 100644
--- a/docs/graphs/gatv2/experiment.html
+++ b/docs/graphs/gatv2/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gatv2/index.html b/docs/graphs/gatv2/index.html
index e5121cdd..d07eb432 100644
--- a/docs/graphs/gatv2/index.html
+++ b/docs/graphs/gatv2/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/gatv2/readme.html b/docs/graphs/gatv2/readme.html
index 1add2276..46e307be 100644
--- a/docs/graphs/gatv2/readme.html
+++ b/docs/graphs/gatv2/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/graphs/index.html b/docs/graphs/index.html
index c5827493..b1674934 100644
--- a/docs/graphs/index.html
+++ b/docs/graphs/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/hypernetworks/experiment.html b/docs/hypernetworks/experiment.html
index 1d855c41..4853af95 100644
--- a/docs/hypernetworks/experiment.html
+++ b/docs/hypernetworks/experiment.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/hypernetworks/hyper_lstm.html b/docs/hypernetworks/hyper_lstm.html
index 2a9f09cd..185e6c3f 100644
--- a/docs/hypernetworks/hyper_lstm.html
+++ b/docs/hypernetworks/hyper_lstm.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/hypernetworks/index.html b/docs/hypernetworks/index.html
index 99e296cc..cfb06468 100644
--- a/docs/hypernetworks/index.html
+++ b/docs/hypernetworks/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/index.html b/docs/index.html
index b61e110d..37c3a867 100644
--- a/docs/index.html
+++ b/docs/index.html
@@ -50,8 +50,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/lstm/index.html b/docs/lstm/index.html
index 9e940849..20fd98c3 100644
--- a/docs/lstm/index.html
+++ b/docs/lstm/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/batch_channel_norm/index.html b/docs/normalization/batch_channel_norm/index.html
index be916f01..f132e955 100644
--- a/docs/normalization/batch_channel_norm/index.html
+++ b/docs/normalization/batch_channel_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/batch_norm/cifar10.html b/docs/normalization/batch_norm/cifar10.html
index aac40e00..211ef380 100644
--- a/docs/normalization/batch_norm/cifar10.html
+++ b/docs/normalization/batch_norm/cifar10.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/batch_norm/index.html b/docs/normalization/batch_norm/index.html
index 0766dbd0..1c371728 100644
--- a/docs/normalization/batch_norm/index.html
+++ b/docs/normalization/batch_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/batch_norm/mnist.html b/docs/normalization/batch_norm/mnist.html
index 99afce2c..bb17a445 100644
--- a/docs/normalization/batch_norm/mnist.html
+++ b/docs/normalization/batch_norm/mnist.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/batch_norm/readme.html b/docs/normalization/batch_norm/readme.html
index 2077788a..fd7ab53c 100644
--- a/docs/normalization/batch_norm/readme.html
+++ b/docs/normalization/batch_norm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/deep_norm/experiment.html b/docs/normalization/deep_norm/experiment.html
index 45b07cee..c6abfa1b 100644
--- a/docs/normalization/deep_norm/experiment.html
+++ b/docs/normalization/deep_norm/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/deep_norm/index.html b/docs/normalization/deep_norm/index.html
index 7d251e22..9c5457f7 100644
--- a/docs/normalization/deep_norm/index.html
+++ b/docs/normalization/deep_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/group_norm/experiment.html b/docs/normalization/group_norm/experiment.html
index f7ae48ae..d3a43dff 100644
--- a/docs/normalization/group_norm/experiment.html
+++ b/docs/normalization/group_norm/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/group_norm/index.html b/docs/normalization/group_norm/index.html
index d10d1861..0eca6df9 100644
--- a/docs/normalization/group_norm/index.html
+++ b/docs/normalization/group_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/group_norm/readme.html b/docs/normalization/group_norm/readme.html
index 96551af1..84015838 100644
--- a/docs/normalization/group_norm/readme.html
+++ b/docs/normalization/group_norm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/index.html b/docs/normalization/index.html
index f83a24ab..a8ea290d 100644
--- a/docs/normalization/index.html
+++ b/docs/normalization/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/instance_norm/experiment.html b/docs/normalization/instance_norm/experiment.html
index 78e9df34..c0f27e45 100644
--- a/docs/normalization/instance_norm/experiment.html
+++ b/docs/normalization/instance_norm/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/instance_norm/index.html b/docs/normalization/instance_norm/index.html
index b3e7011c..61e50953 100644
--- a/docs/normalization/instance_norm/index.html
+++ b/docs/normalization/instance_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/instance_norm/readme.html b/docs/normalization/instance_norm/readme.html
index d427a059..ac0ee610 100644
--- a/docs/normalization/instance_norm/readme.html
+++ b/docs/normalization/instance_norm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/layer_norm/index.html b/docs/normalization/layer_norm/index.html
index b5e73566..12a57ca8 100644
--- a/docs/normalization/layer_norm/index.html
+++ b/docs/normalization/layer_norm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/layer_norm/readme.html b/docs/normalization/layer_norm/readme.html
index cddcbd5c..8ac90dd1 100644
--- a/docs/normalization/layer_norm/readme.html
+++ b/docs/normalization/layer_norm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/weight_standardization/conv2d.html b/docs/normalization/weight_standardization/conv2d.html
index 48f55d22..c2f14784 100644
--- a/docs/normalization/weight_standardization/conv2d.html
+++ b/docs/normalization/weight_standardization/conv2d.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/weight_standardization/experiment.html b/docs/normalization/weight_standardization/experiment.html
index 1f5bfce0..9f55592f 100644
--- a/docs/normalization/weight_standardization/experiment.html
+++ b/docs/normalization/weight_standardization/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/weight_standardization/index.html b/docs/normalization/weight_standardization/index.html
index 8ab25f84..c7c42656 100644
--- a/docs/normalization/weight_standardization/index.html
+++ b/docs/normalization/weight_standardization/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/normalization/weight_standardization/readme.html b/docs/normalization/weight_standardization/readme.html
index aed9e0c6..50fb959e 100644
--- a/docs/normalization/weight_standardization/readme.html
+++ b/docs/normalization/weight_standardization/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/ada_belief.html b/docs/optimizers/ada_belief.html
index a8535030..b1ce591a 100644
--- a/docs/optimizers/ada_belief.html
+++ b/docs/optimizers/ada_belief.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/adam.html b/docs/optimizers/adam.html
index 2d1b62e5..1079ac14 100644
--- a/docs/optimizers/adam.html
+++ b/docs/optimizers/adam.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/adam_warmup.html b/docs/optimizers/adam_warmup.html
index da2d1dd8..14f2fa78 100644
--- a/docs/optimizers/adam_warmup.html
+++ b/docs/optimizers/adam_warmup.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/adam_warmup_cosine_decay.html b/docs/optimizers/adam_warmup_cosine_decay.html
index ea6d0860..06a1147e 100644
--- a/docs/optimizers/adam_warmup_cosine_decay.html
+++ b/docs/optimizers/adam_warmup_cosine_decay.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/amsgrad.html b/docs/optimizers/amsgrad.html
index a0351240..a6750080 100644
--- a/docs/optimizers/amsgrad.html
+++ b/docs/optimizers/amsgrad.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/configs.html b/docs/optimizers/configs.html
index 1f3a6194..5fd3355d 100644
--- a/docs/optimizers/configs.html
+++ b/docs/optimizers/configs.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/index.html b/docs/optimizers/index.html
index 04e14c9c..5a1156c0 100644
--- a/docs/optimizers/index.html
+++ b/docs/optimizers/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/mnist_experiment.html b/docs/optimizers/mnist_experiment.html
index 7b4ea78e..0e527e28 100644
--- a/docs/optimizers/mnist_experiment.html
+++ b/docs/optimizers/mnist_experiment.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/noam.html b/docs/optimizers/noam.html
index 9233dcb0..eb787243 100644
--- a/docs/optimizers/noam.html
+++ b/docs/optimizers/noam.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/performance_test.html b/docs/optimizers/performance_test.html
index 90a57c31..73c9fd2b 100644
--- a/docs/optimizers/performance_test.html
+++ b/docs/optimizers/performance_test.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/radam.html b/docs/optimizers/radam.html
index d4486139..b83f9f67 100644
--- a/docs/optimizers/radam.html
+++ b/docs/optimizers/radam.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/optimizers/readme.html b/docs/optimizers/readme.html
index d27b7ca7..5ea2400c 100644
--- a/docs/optimizers/readme.html
+++ b/docs/optimizers/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/recurrent_highway_networks/index.html b/docs/recurrent_highway_networks/index.html
index 2c854e36..e561c667 100644
--- a/docs/recurrent_highway_networks/index.html
+++ b/docs/recurrent_highway_networks/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/resnet/experiment.html b/docs/resnet/experiment.html
index 259b72ae..c7396df3 100644
--- a/docs/resnet/experiment.html
+++ b/docs/resnet/experiment.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/resnet/index.html b/docs/resnet/index.html
index d9ebb647..f101d660 100644
--- a/docs/resnet/index.html
+++ b/docs/resnet/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/resnet/readme.html b/docs/resnet/readme.html
index 96aa11ee..6e77fbb1 100644
--- a/docs/resnet/readme.html
+++ b/docs/resnet/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/dqn/experiment.html b/docs/rl/dqn/experiment.html
index 7ede52df..c9f8f1ae 100644
--- a/docs/rl/dqn/experiment.html
+++ b/docs/rl/dqn/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/dqn/index.html b/docs/rl/dqn/index.html
index 8a36953a..8495a3b8 100644
--- a/docs/rl/dqn/index.html
+++ b/docs/rl/dqn/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/dqn/model.html b/docs/rl/dqn/model.html
index b205b1c4..47a1151a 100644
--- a/docs/rl/dqn/model.html
+++ b/docs/rl/dqn/model.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/dqn/readme.html b/docs/rl/dqn/readme.html
index be6f029e..55f11c8f 100644
--- a/docs/rl/dqn/readme.html
+++ b/docs/rl/dqn/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/dqn/replay_buffer.html b/docs/rl/dqn/replay_buffer.html
index 2edc744f..46074c88 100644
--- a/docs/rl/dqn/replay_buffer.html
+++ b/docs/rl/dqn/replay_buffer.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/game.html b/docs/rl/game.html
index 13ed023d..29996b51 100644
--- a/docs/rl/game.html
+++ b/docs/rl/game.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/index.html b/docs/rl/index.html
index a13bd313..b12c0bc1 100644
--- a/docs/rl/index.html
+++ b/docs/rl/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/ppo/experiment.html b/docs/rl/ppo/experiment.html
index 5d6853a8..b3cbd046 100644
--- a/docs/rl/ppo/experiment.html
+++ b/docs/rl/ppo/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/ppo/gae.html b/docs/rl/ppo/gae.html
index 823e2803..b03fdfab 100644
--- a/docs/rl/ppo/gae.html
+++ b/docs/rl/ppo/gae.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/ppo/index.html b/docs/rl/ppo/index.html
index c257a96b..968f69dc 100644
--- a/docs/rl/ppo/index.html
+++ b/docs/rl/ppo/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/rl/ppo/readme.html b/docs/rl/ppo/readme.html
index b8a1fd6b..d26d85b1 100644
--- a/docs/rl/ppo/readme.html
+++ b/docs/rl/ppo/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/sketch_rnn/index.html b/docs/sketch_rnn/index.html
index d82e8d70..26acab84 100644
--- a/docs/sketch_rnn/index.html
+++ b/docs/sketch_rnn/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/aft/experiment.html b/docs/transformers/aft/experiment.html
index 1a9b611f..a5b9267d 100644
--- a/docs/transformers/aft/experiment.html
+++ b/docs/transformers/aft/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/aft/index.html b/docs/transformers/aft/index.html
index f9e27ca4..76e58f7e 100644
--- a/docs/transformers/aft/index.html
+++ b/docs/transformers/aft/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/aft/readme.html b/docs/transformers/aft/readme.html
index ddb90152..20f79ded 100644
--- a/docs/transformers/aft/readme.html
+++ b/docs/transformers/aft/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/alibi/experiment.html b/docs/transformers/alibi/experiment.html
index b88fe300..e8272bf8 100644
--- a/docs/transformers/alibi/experiment.html
+++ b/docs/transformers/alibi/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/alibi/index.html b/docs/transformers/alibi/index.html
index 2c4a94d1..2eea5200 100644
--- a/docs/transformers/alibi/index.html
+++ b/docs/transformers/alibi/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/basic/autoregressive_experiment.html b/docs/transformers/basic/autoregressive_experiment.html
index db399958..b25ac36d 100644
--- a/docs/transformers/basic/autoregressive_experiment.html
+++ b/docs/transformers/basic/autoregressive_experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/basic/index.html b/docs/transformers/basic/index.html
index d4cf4968..a0eaaac2 100644
--- a/docs/transformers/basic/index.html
+++ b/docs/transformers/basic/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/compressive/experiment.html b/docs/transformers/compressive/experiment.html
index c8ff0000..43ae903b 100644
--- a/docs/transformers/compressive/experiment.html
+++ b/docs/transformers/compressive/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/compressive/index.html b/docs/transformers/compressive/index.html
index 8d9df326..1f4a4ffa 100644
--- a/docs/transformers/compressive/index.html
+++ b/docs/transformers/compressive/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/compressive/readme.html b/docs/transformers/compressive/readme.html
index 113a6386..ffd36a42 100644
--- a/docs/transformers/compressive/readme.html
+++ b/docs/transformers/compressive/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/configs.html b/docs/transformers/configs.html
index 8cc3dcc9..c0c907cc 100644
--- a/docs/transformers/configs.html
+++ b/docs/transformers/configs.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fast_weights/experiment.html b/docs/transformers/fast_weights/experiment.html
index 2bcbbb7b..6fb19ca7 100644
--- a/docs/transformers/fast_weights/experiment.html
+++ b/docs/transformers/fast_weights/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fast_weights/index.html b/docs/transformers/fast_weights/index.html
index 58ca330e..cc075f00 100644
--- a/docs/transformers/fast_weights/index.html
+++ b/docs/transformers/fast_weights/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fast_weights/readme.html b/docs/transformers/fast_weights/readme.html
index 9361effb..fef6a137 100644
--- a/docs/transformers/fast_weights/readme.html
+++ b/docs/transformers/fast_weights/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fast_weights/token_wise.html b/docs/transformers/fast_weights/token_wise.html
index 3561e6d0..962f1804 100644
--- a/docs/transformers/fast_weights/token_wise.html
+++ b/docs/transformers/fast_weights/token_wise.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/feed_forward.html b/docs/transformers/feed_forward.html
index 11acaba1..5ce102bc 100644
--- a/docs/transformers/feed_forward.html
+++ b/docs/transformers/feed_forward.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/feedback/experiment.html b/docs/transformers/feedback/experiment.html
index 0eb27989..e97efead 100644
--- a/docs/transformers/feedback/experiment.html
+++ b/docs/transformers/feedback/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/feedback/index.html b/docs/transformers/feedback/index.html
index 0e16104f..2f2e6c50 100644
--- a/docs/transformers/feedback/index.html
+++ b/docs/transformers/feedback/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/feedback/readme.html b/docs/transformers/feedback/readme.html
index d18aba21..aa5281a6 100644
--- a/docs/transformers/feedback/readme.html
+++ b/docs/transformers/feedback/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fnet/experiment.html b/docs/transformers/fnet/experiment.html
index bfe00dca..0ef05bbb 100644
--- a/docs/transformers/fnet/experiment.html
+++ b/docs/transformers/fnet/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fnet/index.html b/docs/transformers/fnet/index.html
index 522729a8..acfa7e74 100644
--- a/docs/transformers/fnet/index.html
+++ b/docs/transformers/fnet/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/fnet/readme.html b/docs/transformers/fnet/readme.html
index d8ff1259..0f45c1ed 100644
--- a/docs/transformers/fnet/readme.html
+++ b/docs/transformers/fnet/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/glu_variants/experiment.html b/docs/transformers/glu_variants/experiment.html
index 581c53f1..6c775774 100644
--- a/docs/transformers/glu_variants/experiment.html
+++ b/docs/transformers/glu_variants/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/glu_variants/index.html b/docs/transformers/glu_variants/index.html
index 5c6995b9..9b3b4b9e 100644
--- a/docs/transformers/glu_variants/index.html
+++ b/docs/transformers/glu_variants/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/glu_variants/simple.html b/docs/transformers/glu_variants/simple.html
index 739a4806..794b5ead 100644
--- a/docs/transformers/glu_variants/simple.html
+++ b/docs/transformers/glu_variants/simple.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/gmlp/experiment.html b/docs/transformers/gmlp/experiment.html
index ea180877..e8a4551d 100644
--- a/docs/transformers/gmlp/experiment.html
+++ b/docs/transformers/gmlp/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/gmlp/index.html b/docs/transformers/gmlp/index.html
index 05ecaf32..8582cc1e 100644
--- a/docs/transformers/gmlp/index.html
+++ b/docs/transformers/gmlp/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/gmlp/readme.html b/docs/transformers/gmlp/readme.html
index b4bfb85d..cd3adb56 100644
--- a/docs/transformers/gmlp/readme.html
+++ b/docs/transformers/gmlp/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/gpt/index.html b/docs/transformers/gpt/index.html
index 69c61e71..eb7a3ee3 100644
--- a/docs/transformers/gpt/index.html
+++ b/docs/transformers/gpt/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/hour_glass/experiment.html b/docs/transformers/hour_glass/experiment.html
index 0dd65313..8df9b1b3 100644
--- a/docs/transformers/hour_glass/experiment.html
+++ b/docs/transformers/hour_glass/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/hour_glass/index.html b/docs/transformers/hour_glass/index.html
index 90b7dbc1..6ab8b301 100644
--- a/docs/transformers/hour_glass/index.html
+++ b/docs/transformers/hour_glass/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/index.html b/docs/transformers/index.html
index 9d749e38..f4e2f7c8 100644
--- a/docs/transformers/index.html
+++ b/docs/transformers/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/knn/build_index.html b/docs/transformers/knn/build_index.html
index 9ebe6de9..ef78152d 100644
--- a/docs/transformers/knn/build_index.html
+++ b/docs/transformers/knn/build_index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/knn/eval_knn.html b/docs/transformers/knn/eval_knn.html
index c6dd870b..438fede8 100644
--- a/docs/transformers/knn/eval_knn.html
+++ b/docs/transformers/knn/eval_knn.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/knn/index.html b/docs/transformers/knn/index.html
index 3d66909c..290d033c 100644
--- a/docs/transformers/knn/index.html
+++ b/docs/transformers/knn/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/knn/train_model.html b/docs/transformers/knn/train_model.html
index 4c1cf015..56e1578a 100644
--- a/docs/transformers/knn/train_model.html
+++ b/docs/transformers/knn/train_model.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/label_smoothing_loss.html b/docs/transformers/label_smoothing_loss.html
index 2083a122..d8f77e12 100644
--- a/docs/transformers/label_smoothing_loss.html
+++ b/docs/transformers/label_smoothing_loss.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mha.html b/docs/transformers/mha.html
index b6d5123c..0480b7d1 100644
--- a/docs/transformers/mha.html
+++ b/docs/transformers/mha.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlm/experiment.html b/docs/transformers/mlm/experiment.html
index f67a8c14..8c5a7dc5 100644
--- a/docs/transformers/mlm/experiment.html
+++ b/docs/transformers/mlm/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlm/index.html b/docs/transformers/mlm/index.html
index 8bffe887..a7e076f1 100644
--- a/docs/transformers/mlm/index.html
+++ b/docs/transformers/mlm/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlm/readme.html b/docs/transformers/mlm/readme.html
index b706e5b6..b205ede2 100644
--- a/docs/transformers/mlm/readme.html
+++ b/docs/transformers/mlm/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlp_mixer/experiment.html b/docs/transformers/mlp_mixer/experiment.html
index 1ba93f01..95d7a250 100644
--- a/docs/transformers/mlp_mixer/experiment.html
+++ b/docs/transformers/mlp_mixer/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlp_mixer/index.html b/docs/transformers/mlp_mixer/index.html
index b03ceb3b..b5c05efe 100644
--- a/docs/transformers/mlp_mixer/index.html
+++ b/docs/transformers/mlp_mixer/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/mlp_mixer/readme.html b/docs/transformers/mlp_mixer/readme.html
index e86082f0..d5f5453c 100644
--- a/docs/transformers/mlp_mixer/readme.html
+++ b/docs/transformers/mlp_mixer/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/models.html b/docs/transformers/models.html
index 691e712e..b4547c8d 100644
--- a/docs/transformers/models.html
+++ b/docs/transformers/models.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/positional_encoding.html b/docs/transformers/positional_encoding.html
index fa6e2142..b088e591 100644
--- a/docs/transformers/positional_encoding.html
+++ b/docs/transformers/positional_encoding.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/primer_ez/efficient.html b/docs/transformers/primer_ez/efficient.html
index 475bf334..d5d6bfeb 100644
--- a/docs/transformers/primer_ez/efficient.html
+++ b/docs/transformers/primer_ez/efficient.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/primer_ez/experiment.html b/docs/transformers/primer_ez/experiment.html
index 5f69856e..494908a2 100644
--- a/docs/transformers/primer_ez/experiment.html
+++ b/docs/transformers/primer_ez/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/primer_ez/index.html b/docs/transformers/primer_ez/index.html
index a08455ed..8d6e6a79 100644
--- a/docs/transformers/primer_ez/index.html
+++ b/docs/transformers/primer_ez/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/primer_ez/readme.html b/docs/transformers/primer_ez/readme.html
index 3e4b13e7..949e09d0 100644
--- a/docs/transformers/primer_ez/readme.html
+++ b/docs/transformers/primer_ez/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/primer_ez/variations.html b/docs/transformers/primer_ez/variations.html
index e103e922..e80733b3 100644
--- a/docs/transformers/primer_ez/variations.html
+++ b/docs/transformers/primer_ez/variations.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/relative_mha.html b/docs/transformers/relative_mha.html
index 1a95f0ee..1c60ee0b 100644
--- a/docs/transformers/relative_mha.html
+++ b/docs/transformers/relative_mha.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/bert_embeddings.html b/docs/transformers/retro/bert_embeddings.html
index 456e3e2f..4f9d30de 100644
--- a/docs/transformers/retro/bert_embeddings.html
+++ b/docs/transformers/retro/bert_embeddings.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/database.html b/docs/transformers/retro/database.html
index 72388be1..51e46d99 100644
--- a/docs/transformers/retro/database.html
+++ b/docs/transformers/retro/database.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/dataset.html b/docs/transformers/retro/dataset.html
index aa9a64ff..e6ae1f51 100644
--- a/docs/transformers/retro/dataset.html
+++ b/docs/transformers/retro/dataset.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/index.html b/docs/transformers/retro/index.html
index 5466d2d7..6bdaa852 100644
--- a/docs/transformers/retro/index.html
+++ b/docs/transformers/retro/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/model.html b/docs/transformers/retro/model.html
index 19e9ce09..c06690b6 100644
--- a/docs/transformers/retro/model.html
+++ b/docs/transformers/retro/model.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/retro/train.html b/docs/transformers/retro/train.html
index 9b27baf6..227df911 100644
--- a/docs/transformers/retro/train.html
+++ b/docs/transformers/retro/train.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/rope/experiment.html b/docs/transformers/rope/experiment.html
index 0174cd2f..c6a382d1 100644
--- a/docs/transformers/rope/experiment.html
+++ b/docs/transformers/rope/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/rope/index.html b/docs/transformers/rope/index.html
index fea636d4..6f0a80b1 100644
--- a/docs/transformers/rope/index.html
+++ b/docs/transformers/rope/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/rope/value_pe/arithmetic_experiment.html b/docs/transformers/rope/value_pe/arithmetic_experiment.html
index 08c66feb..12398e01 100644
--- a/docs/transformers/rope/value_pe/arithmetic_experiment.html
+++ b/docs/transformers/rope/value_pe/arithmetic_experiment.html
@@ -53,8 +53,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/rope/value_pe/experiment.html b/docs/transformers/rope/value_pe/experiment.html
index f3ae4140..90932413 100644
--- a/docs/transformers/rope/value_pe/experiment.html
+++ b/docs/transformers/rope/value_pe/experiment.html
@@ -53,8 +53,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/rope/value_pe/index.html b/docs/transformers/rope/value_pe/index.html
index a4a19709..fbaba190 100644
--- a/docs/transformers/rope/value_pe/index.html
+++ b/docs/transformers/rope/value_pe/index.html
@@ -53,8 +53,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/switch/experiment.html b/docs/transformers/switch/experiment.html
index c0bcadd2..0d091120 100644
--- a/docs/transformers/switch/experiment.html
+++ b/docs/transformers/switch/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/switch/index.html b/docs/transformers/switch/index.html
index 74d7eb77..99e16260 100644
--- a/docs/transformers/switch/index.html
+++ b/docs/transformers/switch/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/switch/readme.html b/docs/transformers/switch/readme.html
index 08997fc4..31dedc63 100644
--- a/docs/transformers/switch/readme.html
+++ b/docs/transformers/switch/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/utils.html b/docs/transformers/utils.html
index 610af833..f03a78f4 100644
--- a/docs/transformers/utils.html
+++ b/docs/transformers/utils.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/vit/experiment.html b/docs/transformers/vit/experiment.html
index f9232ca1..3dfe3e73 100644
--- a/docs/transformers/vit/experiment.html
+++ b/docs/transformers/vit/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/vit/index.html b/docs/transformers/vit/index.html
index 842a5713..3e291a7c 100644
--- a/docs/transformers/vit/index.html
+++ b/docs/transformers/vit/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/vit/readme.html b/docs/transformers/vit/readme.html
index 868ef3d4..db6c82e5 100644
--- a/docs/transformers/vit/readme.html
+++ b/docs/transformers/vit/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/xl/experiment.html b/docs/transformers/xl/experiment.html
index 79436aa8..f5388de1 100644
--- a/docs/transformers/xl/experiment.html
+++ b/docs/transformers/xl/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/xl/index.html b/docs/transformers/xl/index.html
index 2871e1f8..be1e5135 100644
--- a/docs/transformers/xl/index.html
+++ b/docs/transformers/xl/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/xl/readme.html b/docs/transformers/xl/readme.html
index 51fd99d8..aad39161 100644
--- a/docs/transformers/xl/readme.html
+++ b/docs/transformers/xl/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/transformers/xl/relative_mha.html b/docs/transformers/xl/relative_mha.html
index b758985a..33269443 100644
--- a/docs/transformers/xl/relative_mha.html
+++ b/docs/transformers/xl/relative_mha.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/uncertainty/evidence/experiment.html b/docs/uncertainty/evidence/experiment.html
index 0a15b426..9af2be83 100644
--- a/docs/uncertainty/evidence/experiment.html
+++ b/docs/uncertainty/evidence/experiment.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/uncertainty/evidence/index.html b/docs/uncertainty/evidence/index.html
index b0bc3bdc..4f8824f4 100644
--- a/docs/uncertainty/evidence/index.html
+++ b/docs/uncertainty/evidence/index.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/uncertainty/evidence/readme.html b/docs/uncertainty/evidence/readme.html
index d14d6558..3aa3010d 100644
--- a/docs/uncertainty/evidence/readme.html
+++ b/docs/uncertainty/evidence/readme.html
@@ -52,8 +52,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/uncertainty/index.html b/docs/uncertainty/index.html
index f20976cc..94af4c96 100644
--- a/docs/uncertainty/index.html
+++ b/docs/uncertainty/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/uncertainty/readme.html b/docs/uncertainty/readme.html
index bad2251e..4cb20dd5 100644
--- a/docs/uncertainty/readme.html
+++ b/docs/uncertainty/readme.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/utils/index.html b/docs/utils/index.html
index dd8ee6e4..19ceeee9 100644
--- a/docs/utils/index.html
+++ b/docs/utils/index.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/docs/utils/tokenizer.html b/docs/utils/tokenizer.html
index 773ede99..fbf5a372 100644
--- a/docs/utils/tokenizer.html
+++ b/docs/utils/tokenizer.html
@@ -51,8 +51,8 @@
             </p>
             <p>
                 <a href="https://github.com/sponsors/labmlai" target="_blank">
-                    <img alt="Github Sponsors"
-                         src="https://img.shields.io/github/sponsors/labmlai?label=github%20sponsors"
+                    <img alt="Sponsor"
+                         src="https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86"
                          style="max-width:100%;"/></a>
                 <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
                     <img alt="Github"
diff --git a/readme.md b/readme.md
index c57e5ac5..f3a906cc 100644
--- a/readme.md
+++ b/readme.md
@@ -1,4 +1,5 @@
 [![Twitter](https://img.shields.io/twitter/follow/labmlai?style=social)](https://twitter.com/labmlai)
+[![Sponsor](https://img.shields.io/static/v1?label=Sponsor&message=%E2%9D%A4&logo=GitHub&color=%23fe8e86)](https://github.com/sponsors/labmlai)
 
 # [labml.ai Deep Learning Paper Implementations](https://nn.labml.ai/index.html)