DouglasOrr published a site update

graphcore-research · Jul 24, 2024 · 056ffde · 056ffde
1 parent 34641c4
commit 056ffde
Show file tree

Hide file tree

Showing 175 changed files with 28,269 additions and 2,133 deletions.
diff --git a/_modules/collections.html b/_modules/collections.html
diff --git a/_modules/index.html b/_modules/index.html
@@ -31,7 +31,7 @@
             unit-scaling
           </a>
               <div class="version">
-                0.1
+                0.2
               </div>
 <div role="search">
   <form id="rtd-search-form" class="wy-form" action="../search.html" method="get">
@@ -74,12 +74,21 @@
            <div itemprop="articleBody">
 
   <h1>All modules for which code is available</h1>
-<ul><li><a href="torch/nn/modules/sparse.html">torch.nn.modules.sparse</a></li>
+<ul><li><a href="collections.html">collections</a></li>
+<li><a href="torch/_tensor.html">torch._tensor</a></li>
+<li><a href="torch/nn/modules/sparse.html">torch.nn.modules.sparse</a></li>
+<li><a href="torch/optim/adam.html">torch.optim.adam</a></li>
+<li><a href="torch/optim/adamw.html">torch.optim.adamw</a></li>
+<li><a href="torch/optim/sgd.html">torch.optim.sgd</a></li>
+<li><a href="typing.html">typing</a></li>
 <li><a href="unit_scaling/_modules.html">unit_scaling._modules</a></li>
 <li><a href="unit_scaling/analysis.html">unit_scaling.analysis</a></li>
 <li><a href="unit_scaling/constraints.html">unit_scaling.constraints</a></li>
+<li><a href="unit_scaling/core/functional.html">unit_scaling.core.functional</a></li>
 <li><a href="unit_scaling/formats.html">unit_scaling.formats</a></li>
 <li><a href="unit_scaling/functional.html">unit_scaling.functional</a></li>
+<li><a href="unit_scaling/optim.html">unit_scaling.optim</a></li>
+<li><a href="unit_scaling/parameter.html">unit_scaling.parameter</a></li>
 <li><a href="unit_scaling/scale.html">unit_scaling.scale</a></li>
 <li><a href="unit_scaling/transforms/_compile.html">unit_scaling.transforms._compile</a></li>
 <li><a href="unit_scaling/transforms/_simulate_format.html">unit_scaling.transforms._simulate_format</a></li>

diff --git a/_modules/torch/_tensor.html b/_modules/torch/_tensor.html
diff --git a/_modules/torch/nn/modules/sparse.html b/_modules/torch/nn/modules/sparse.html
@@ -31,7 +31,7 @@
             unit-scaling
           </a>
               <div class="version">
-                0.1
+                0.2
               </div>
 <div role="search">
   <form id="rtd-search-form" class="wy-form" action="../../../../search.html" method="get">
@@ -75,7 +75,8 @@
            <div itemprop="articleBody">
 
   <h1>Source code for torch.nn.modules.sparse</h1><div class="highlight"><pre>
-<span></span><span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Optional</span>
+<span></span><span class="c1"># mypy: allow-untyped-defs</span>
+<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Optional</span>
 
 <span class="kn">import</span> <span class="nn">torch</span>
 <span class="kn">from</span> <span class="nn">torch</span> <span class="kn">import</span> <span class="n">Tensor</span>

diff --git a/_modules/torch/optim/adam.html b/_modules/torch/optim/adam.html
diff --git a/_modules/torch/optim/adamw.html b/_modules/torch/optim/adamw.html
diff --git a/_modules/torch/optim/sgd.html b/_modules/torch/optim/sgd.html
diff --git a/_modules/typing.html b/_modules/typing.html
diff --git a/_modules/unit_scaling/_modules.html b/_modules/unit_scaling/_modules.html
diff --git a/_modules/unit_scaling/analysis.html b/_modules/unit_scaling/analysis.html
@@ -31,7 +31,7 @@
             unit-scaling
           </a>
               <div class="version">
-                0.1
+                0.2
               </div>
 <div role="search">
   <form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">

diff --git a/_modules/unit_scaling/constraints.html b/_modules/unit_scaling/constraints.html
@@ -31,7 +31,7 @@
             unit-scaling
           </a>
               <div class="version">
-                0.1
+                0.2
               </div>
 <div role="search">
   <form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">

diff --git a/_modules/unit_scaling/core/functional.html b/_modules/unit_scaling/core/functional.html
diff --git a/_modules/unit_scaling/formats.html b/_modules/unit_scaling/formats.html
@@ -31,7 +31,7 @@
             unit-scaling
           </a>
               <div class="version">
-                0.1
+                0.2
               </div>
 <div role="search">
   <form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
@@ -195,9 +195,7 @@ <h1>Source code for unit_scaling.formats</h1><div class="highlight"><pre>
 
         <span class="k">class</span> <span class="nc">QuantiseForward</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">Function</span><span class="p">):</span>
             <span class="nd">@staticmethod</span>
-            <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span>  <span class="c1"># type:ignore[override]</span>
-                <span class="n">ctx</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">function</span><span class="o">.</span><span class="n">FunctionCtx</span><span class="p">,</span> <span class="n">x</span><span class="p">:</span> <span class="n">Tensor</span>
-            <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
+            <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="n">ctx</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">function</span><span class="o">.</span><span class="n">FunctionCtx</span><span class="p">,</span> <span class="n">x</span><span class="p">:</span> <span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
                 <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">quantise</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
 
             <span class="nd">@staticmethod</span>
@@ -220,9 +218,7 @@ <h1>Source code for unit_scaling.formats</h1><div class="highlight"><pre>
 
         <span class="k">class</span> <span class="nc">QuantiseBackward</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">Function</span><span class="p">):</span>
             <span class="nd">@staticmethod</span>
-            <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span>  <span class="c1"># type:ignore[override]</span>
-                <span class="n">ctx</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">function</span><span class="o">.</span><span class="n">FunctionCtx</span><span class="p">,</span> <span class="n">x</span><span class="p">:</span> <span class="n">Tensor</span>
-            <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
+            <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="n">ctx</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">autograd</span><span class="o">.</span><span class="n">function</span><span class="o">.</span><span class="n">FunctionCtx</span><span class="p">,</span> <span class="n">x</span><span class="p">:</span> <span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
                 <span class="k">return</span> <span class="n">x</span>
 
             <span class="nd">@staticmethod</span>