Improved JAX Scalify quickstart Jupyter notebook.

In-depth explanations and examples.
graphcore-research · Jun 17, 2024 · f19f6b0 · f19f6b0
1 parent 41f209e
commit f19f6b0
Show file tree

Hide file tree

Showing 3 changed files with 673 additions and 270 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# JAX Scalify: end-to-end scaled Arithmetics
+# JAX Scalify: end-to-end scaled arithmetic
 
 **JAX Scalify** is a thin library implementing numerically stable scaled arithmetics, allowing easy training and inference of
 deep neural networks in low precision (BF16, FP16, FP8).

diff --git a/examples/autoscale-quickstart.ipynb b/examples/autoscale-quickstart.ipynb