diff --git a/docs/.buildinfo b/docs/.buildinfo
index 97d1369..5a1f65a 100644
--- a/docs/.buildinfo
+++ b/docs/.buildinfo
@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: d4048482bcbca2e55b6abcf6a3654224
+config: 869c5f05717ef3f860009eb902a631ec
 tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/docs/_modules/index.html b/docs/_modules/index.html
index d30574f..897174e 100644
--- a/docs/_modules/index.html
+++ b/docs/_modules/index.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Overview: module code &mdash; nets 0.1 documentation</title>
+  <title>Overview: module code &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -363,7 +363,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/autograd/functions.html b/docs/_modules/nets/autograd/functions.html
index e0aed58..802736e 100644
--- a/docs/_modules/nets/autograd/functions.html
+++ b/docs/_modules/nets/autograd/functions.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.autograd.functions &mdash; nets 0.1 documentation</title>
+  <title>nets.autograd.functions &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -742,7 +742,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/autograd/hook.html b/docs/_modules/nets/autograd/hook.html
index 626305b..045165b 100644
--- a/docs/_modules/nets/autograd/hook.html
+++ b/docs/_modules/nets/autograd/hook.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.autograd.hook &mdash; nets 0.1 documentation</title>
+  <title>nets.autograd.hook &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -349,7 +349,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/autograd/numeric.html b/docs/_modules/nets/autograd/numeric.html
index 5cad98b..2727324 100644
--- a/docs/_modules/nets/autograd/numeric.html
+++ b/docs/_modules/nets/autograd/numeric.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.autograd.numeric &mdash; nets 0.1 documentation</title>
+  <title>nets.autograd.numeric &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -570,7 +570,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/autograd/ops.html b/docs/_modules/nets/autograd/ops.html
index d200746..3d4d014 100644
--- a/docs/_modules/nets/autograd/ops.html
+++ b/docs/_modules/nets/autograd/ops.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.autograd.ops &mdash; nets 0.1 documentation</title>
+  <title>nets.autograd.ops &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -797,7 +797,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/autograd/parameter.html b/docs/_modules/nets/autograd/parameter.html
index 192dd40..dac3a53 100644
--- a/docs/_modules/nets/autograd/parameter.html
+++ b/docs/_modules/nets/autograd/parameter.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.autograd.parameter &mdash; nets 0.1 documentation</title>
+  <title>nets.autograd.parameter &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -453,7 +453,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/data/batch.html b/docs/_modules/nets/data/batch.html
index 7d7503e..5683aa1 100644
--- a/docs/_modules/nets/data/batch.html
+++ b/docs/_modules/nets/data/batch.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.data.batch &mdash; nets 0.1 documentation</title>
+  <title>nets.data.batch &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -353,7 +353,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/data/dataset.html b/docs/_modules/nets/data/dataset.html
index 2b15a92..62bd4f7 100644
--- a/docs/_modules/nets/data/dataset.html
+++ b/docs/_modules/nets/data/dataset.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.data.dataset &mdash; nets 0.1 documentation</title>
+  <title>nets.data.dataset &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -406,7 +406,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/data/example.html b/docs/_modules/nets/data/example.html
index 14acaa3..e279676 100644
--- a/docs/_modules/nets/data/example.html
+++ b/docs/_modules/nets/data/example.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.data.example &mdash; nets 0.1 documentation</title>
+  <title>nets.data.example &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -390,7 +390,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/data/iterator.html b/docs/_modules/nets/data/iterator.html
index 32c319b..af92a61 100644
--- a/docs/_modules/nets/data/iterator.html
+++ b/docs/_modules/nets/data/iterator.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.data.iterator &mdash; nets 0.1 documentation</title>
+  <title>nets.data.iterator &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -365,7 +365,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/datasets/cifar.html b/docs/_modules/nets/datasets/cifar.html
index 6134a15..b5a505c 100644
--- a/docs/_modules/nets/datasets/cifar.html
+++ b/docs/_modules/nets/datasets/cifar.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.datasets.cifar &mdash; nets 0.1 documentation</title>
+  <title>nets.datasets.cifar &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -438,7 +438,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/datasets/mnist.html b/docs/_modules/nets/datasets/mnist.html
index 306f9d9..20da01e 100644
--- a/docs/_modules/nets/datasets/mnist.html
+++ b/docs/_modules/nets/datasets/mnist.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.datasets.mnist &mdash; nets 0.1 documentation</title>
+  <title>nets.datasets.mnist &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -415,7 +415,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/activation.html b/docs/_modules/nets/nn/activation.html
index 6df099d..fa97dd3 100644
--- a/docs/_modules/nets/nn/activation.html
+++ b/docs/_modules/nets/nn/activation.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.activation &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.activation &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -544,7 +544,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/functional.html b/docs/_modules/nets/nn/functional.html
index a3a48e9..0bfa6ee 100644
--- a/docs/_modules/nets/nn/functional.html
+++ b/docs/_modules/nets/nn/functional.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.functional &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.functional &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -373,7 +373,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/loss.html b/docs/_modules/nets/nn/loss.html
index b5607b8..080749b 100644
--- a/docs/_modules/nets/nn/loss.html
+++ b/docs/_modules/nets/nn/loss.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.loss &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.loss &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -423,7 +423,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/conv.html b/docs/_modules/nets/nn/modules/conv.html
index bd31ba5..33e9fd9 100644
--- a/docs/_modules/nets/nn/modules/conv.html
+++ b/docs/_modules/nets/nn/modules/conv.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.conv &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.conv &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -412,7 +412,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/dnn.html b/docs/_modules/nets/nn/modules/dnn.html
index 4960386..522540d 100644
--- a/docs/_modules/nets/nn/modules/dnn.html
+++ b/docs/_modules/nets/nn/modules/dnn.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.dnn &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.dnn &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -505,7 +505,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/dropout.html b/docs/_modules/nets/nn/modules/dropout.html
index f253b5d..1ebe631 100644
--- a/docs/_modules/nets/nn/modules/dropout.html
+++ b/docs/_modules/nets/nn/modules/dropout.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.dropout &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.dropout &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -358,7 +358,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/linear.html b/docs/_modules/nets/nn/modules/linear.html
index 1fd9603..517d565 100644
--- a/docs/_modules/nets/nn/modules/linear.html
+++ b/docs/_modules/nets/nn/modules/linear.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.linear &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.linear &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -429,7 +429,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/module.html b/docs/_modules/nets/nn/modules/module.html
index 87449ce..d9193f9 100644
--- a/docs/_modules/nets/nn/modules/module.html
+++ b/docs/_modules/nets/nn/modules/module.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.module &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.module &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -491,7 +491,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/pool.html b/docs/_modules/nets/nn/modules/pool.html
index 97a355d..cda4800 100644
--- a/docs/_modules/nets/nn/modules/pool.html
+++ b/docs/_modules/nets/nn/modules/pool.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.pool &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.pool &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -398,7 +398,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/rnn.html b/docs/_modules/nets/nn/modules/rnn.html
index d4d00c4..4ba6cf2 100644
--- a/docs/_modules/nets/nn/modules/rnn.html
+++ b/docs/_modules/nets/nn/modules/rnn.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.rnn &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.rnn &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -468,7 +468,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/rnnbase.html b/docs/_modules/nets/nn/modules/rnnbase.html
index d1075e9..486743e 100644
--- a/docs/_modules/nets/nn/modules/rnnbase.html
+++ b/docs/_modules/nets/nn/modules/rnnbase.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.rnnbase &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.rnnbase &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -353,7 +353,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/modules/sequential.html b/docs/_modules/nets/nn/modules/sequential.html
index 573b18c..e25ea75 100644
--- a/docs/_modules/nets/nn/modules/sequential.html
+++ b/docs/_modules/nets/nn/modules/sequential.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.modules.sequential &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.modules.sequential &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -374,7 +374,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/nn/utils.html b/docs/_modules/nets/nn/utils.html
index f8c357a..44ad063 100644
--- a/docs/_modules/nets/nn/utils.html
+++ b/docs/_modules/nets/nn/utils.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.nn.utils &mdash; nets 0.1 documentation</title>
+  <title>nets.nn.utils &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -356,7 +356,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/optim/adagrad.html b/docs/_modules/nets/optim/adagrad.html
index 28220b6..9f43ebb 100644
--- a/docs/_modules/nets/optim/adagrad.html
+++ b/docs/_modules/nets/optim/adagrad.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.optim.adagrad &mdash; nets 0.1 documentation</title>
+  <title>nets.optim.adagrad &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -354,7 +354,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/optim/adam.html b/docs/_modules/nets/optim/adam.html
index 99cb688..be5b266 100644
--- a/docs/_modules/nets/optim/adam.html
+++ b/docs/_modules/nets/optim/adam.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.optim.adam &mdash; nets 0.1 documentation</title>
+  <title>nets.optim.adam &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -368,7 +368,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/optim/optimizer.html b/docs/_modules/nets/optim/optimizer.html
index 7a06239..da374af 100644
--- a/docs/_modules/nets/optim/optimizer.html
+++ b/docs/_modules/nets/optim/optimizer.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.optim.optimizer &mdash; nets 0.1 documentation</title>
+  <title>nets.optim.optimizer &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -363,7 +363,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/optim/rmsprop.html b/docs/_modules/nets/optim/rmsprop.html
index de675e5..d10a78d 100644
--- a/docs/_modules/nets/optim/rmsprop.html
+++ b/docs/_modules/nets/optim/rmsprop.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.optim.rmsprop &mdash; nets 0.1 documentation</title>
+  <title>nets.optim.rmsprop &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -363,7 +363,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/optim/sgd.html b/docs/_modules/nets/optim/sgd.html
index bbb8363..ec43851 100644
--- a/docs/_modules/nets/optim/sgd.html
+++ b/docs/_modules/nets/optim/sgd.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.optim.sgd &mdash; nets 0.1 documentation</title>
+  <title>nets.optim.sgd &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -367,7 +367,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/solver/solver.html b/docs/_modules/nets/solver/solver.html
index fb3fe40..0b28b7b 100644
--- a/docs/_modules/nets/solver/solver.html
+++ b/docs/_modules/nets/solver/solver.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.solver.solver &mdash; nets 0.1 documentation</title>
+  <title>nets.solver.solver &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -573,7 +573,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/tensor.html b/docs/_modules/nets/tensor.html
index 2ff921a..96fa377 100644
--- a/docs/_modules/nets/tensor.html
+++ b/docs/_modules/nets/tensor.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.tensor &mdash; nets 0.1 documentation</title>
+  <title>nets.tensor &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -714,7 +714,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/utils/display.html b/docs/_modules/nets/utils/display.html
index 29505ca..90d6135 100644
--- a/docs/_modules/nets/utils/display.html
+++ b/docs/_modules/nets/utils/display.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.utils.display &mdash; nets 0.1 documentation</title>
+  <title>nets.utils.display &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -410,7 +410,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/utils/errors.html b/docs/_modules/nets/utils/errors.html
index 9025be4..1a1f1a9 100644
--- a/docs/_modules/nets/utils/errors.html
+++ b/docs/_modules/nets/utils/errors.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.utils.errors &mdash; nets 0.1 documentation</title>
+  <title>nets.utils.errors &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -342,7 +342,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_modules/nets/utils/functions.html b/docs/_modules/nets/utils/functions.html
index 05ce92e..a4d0347 100644
--- a/docs/_modules/nets/utils/functions.html
+++ b/docs/_modules/nets/utils/functions.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets.utils.functions &mdash; nets 0.1 documentation</title>
+  <title>nets.utils.functions &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -371,7 +371,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/_static/documentation_options.js b/docs/_static/documentation_options.js
index 8b17e7d..2a5dd51 100644
--- a/docs/_static/documentation_options.js
+++ b/docs/_static/documentation_options.js
@@ -1,6 +1,6 @@
 var DOCUMENTATION_OPTIONS = {
     URL_ROOT: document.getElementById("documentation_options").getAttribute('data-url_root'),
-    VERSION: '0.1',
+    VERSION: '0.0.2',
     LANGUAGE: 'None',
     COLLAPSE_INDEX: false,
     BUILDER: 'html',
diff --git a/docs/genindex.html b/docs/genindex.html
index 3580d9d..bd30c8f 100644
--- a/docs/genindex.html
+++ b/docs/genindex.html
@@ -10,7 +10,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Index &mdash; nets 0.1 documentation</title>
+  <title>Index &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -966,7 +966,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/index.html b/docs/index.html
index 41cd82c..fac9374 100644
--- a/docs/index.html
+++ b/docs/index.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Welcome to NETS documentation! &mdash; nets 0.1 documentation</title>
+  <title>Welcome to NETS documentation! &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -389,7 +389,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/py-modindex.html b/docs/py-modindex.html
index 6f424e0..3d5c76c 100644
--- a/docs/py-modindex.html
+++ b/docs/py-modindex.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Python Module Index &mdash; nets 0.1 documentation</title>
+  <title>Python Module Index &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -534,7 +534,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/search.html b/docs/search.html
index d750547..cdf93d9 100644
--- a/docs/search.html
+++ b/docs/search.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Search &mdash; nets 0.1 documentation</title>
+  <title>Search &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -346,7 +346,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/source/getting-started.html b/docs/source/getting-started.html
index 2f3e5d3..ea175c6 100644
--- a/docs/source/getting-started.html
+++ b/docs/source/getting-started.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>About &mdash; nets 0.1 documentation</title>
+  <title>About &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -442,7 +442,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/source/package.html b/docs/source/package.html
index 716254b..9e2b538 100644
--- a/docs/source/package.html
+++ b/docs/source/package.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>nets &mdash; nets 0.1 documentation</title>
+  <title>nets &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -3097,7 +3097,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/source/tutorial-cnn.html b/docs/source/tutorial-cnn.html
index 9452324..3fee01e 100644
--- a/docs/source/tutorial-cnn.html
+++ b/docs/source/tutorial-cnn.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Convolutional Layers &mdash; nets 0.1 documentation</title>
+  <title>Convolutional Layers &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -612,7 +612,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/source/tutorial-xor.html b/docs/source/tutorial-xor.html
index f85e78a..8cfa65d 100644
--- a/docs/source/tutorial-xor.html
+++ b/docs/source/tutorial-xor.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>XOR &mdash; nets 0.1 documentation</title>
+  <title>XOR &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -487,7 +487,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>
diff --git a/docs/source/tutorial.html b/docs/source/tutorial.html
index 67643ea..0f6a1b9 100644
--- a/docs/source/tutorial.html
+++ b/docs/source/tutorial.html
@@ -9,7 +9,7 @@
   
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
   
-  <title>Tensor &mdash; nets 0.1 documentation</title>
+  <title>Tensor &mdash; nets 0.0.2 documentation</title>
   
 
   
@@ -658,7 +658,7 @@ <h2>Resources</h2>
       <div class="footer-links-wrapper">
         <div class="footer-links-col">
           <ul>
-            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">PyTorch</a></li>
+            <li class="list-title"><a href="https://arthurdjn.github.io/nets/">NETS</a></li>
             <li><a href="https://arthurdjn.github.io/nets/source/getting-started.html">Get Started</a></li>
             <li><a href="https://arthurdujardin.com/project/nets.html">Blog</a></li>
           </ul>