From 26d3ccf8a7cce888c7865de0e4543a252a681993 Mon Sep 17 00:00:00 2001
From: mjwen <wenxx151@gmail.com>
Date: Sat, 20 Nov 2021 22:39:34 -0800
Subject: [PATCH] Update version to 0.3.1 and generate docs

---
 .../auto_examples/auto_examples_jupyter.zip   | Bin 27875 -> 27953 bytes
 .../auto_examples/auto_examples_python.zip    | Bin 20902 -> 20981 bytes
 docs/source/auto_examples/example_nn_Si.ipynb |   4 +-
 docs/source/auto_examples/example_nn_Si.py    |   4 +-
 .../source/auto_examples/example_nn_Si.py.md5 |   2 +-
 docs/source/auto_examples/example_nn_Si.rst   |  47 ++++++++++--------
 .../source/auto_examples/example_nn_SiC.ipynb |   2 +-
 docs/source/auto_examples/example_nn_SiC.py   |   2 +-
 .../auto_examples/example_nn_SiC.py.md5       |   2 +-
 docs/source/auto_examples/example_nn_SiC.rst  |  13 +----
 .../example_nn_SiC_codeobj.pickle             | Bin 10227 -> 10218 bytes
 .../example_nn_Si_codeobj.pickle              | Bin 8397 -> 8388 bytes
 .../auto_examples/sg_execution_times.rst      |  10 ++--
 docs/source/conf.py                           |   2 +-
 14 files changed, 42 insertions(+), 46 deletions(-)

diff --git a/docs/source/auto_examples/auto_examples_jupyter.zip b/docs/source/auto_examples/auto_examples_jupyter.zip
index d4bc9d4c56f0e157016ff2caab3d79705cf42659..c438931648857a6fffad853f1086e16bcf23fa50 100644
GIT binary patch
delta 235
zcmaESlX2rMM)3e|W)=|!5D3~*65M-{_x>qf28L4`#kX;A>nNlbl-jx_<`kz+ewd)K
z`76hI2i^dnvU|;CYo9AJFg)M9(X&a^EF@o{q$n{nPa!`Kq(eucIJHC}ApxX0q^LAC
zAwdBsmyqC`n3G(ZlUR~pl#rkolegJ0egh-29h)B}%;MrjamM71>|n+}lka5vFx?fO
p{7zhUvU`pequS)695s0qhX5TT4E6<tSkkz3^7b4xw!PUPJ^((xR5}0v

delta 163
zcmdmZi}CSIM)3e|W)=|!5U9vu4h}rKZQXue28R6`#kX-x)=$*f#JS#KvVj5%=c9oC
zR!bBa7?y0_<JlxSxkY{VWJjTIn~f5ubMd0+oO~@inDOIetsEbwOX8D%ipx%(nxn-i
pJ^4V6nu6p`?qIk<K*NN=7DI?7jk7!_AM_NRES)RBmY)q01OR=sHW2^-

diff --git a/docs/source/auto_examples/auto_examples_python.zip b/docs/source/auto_examples/auto_examples_python.zip
index d709b32546f0436f73dab5f62675e20422853446..7e7c4ce6a6c0c994f9e182bfb6e6695938a71a6e 100644
GIT binary patch
delta 258
zcmZ3snDOgkM)3e|W)=|!5D45-5?pb$@4O5r1B1*)@ndY<Itu9prM7N~ImM}yCum7-
z{>H|o&ua!$R+{Hsk|@Q%khq!GEL6xsSs^4}p`<7=GfyEu52Qy&p*XcfAt3>zJ*22K
zH6cL(D3_4noS2hbnv+<PUzCub$E7?u(c=}mC6m9g32ZL$yvfFk;-1M-0RfEOlUE0L
zGOg#Ee1K1CvOu5~lb!Hne_=750B=Sn5oU-lP@D&Jq5#-$5MoIqquJ!TKqa;b0U!YY
DV~<O7

delta 180
zcmeymm~q)+M)3e|W)=|!5U9vu4(6z-UL(lKz#zC${21Hh*;<m5&+-UNE)kNP?4ZxL
z*@FGD9<St1?%+43wNquK7#L(XOPGZUO=c4Goh-vMZ}MC%aiIFV&5B;tY`iGOO<oxg
z!00gfSAZwe628f&`J^Vt2Wl}IPwom-RzNWjXs7_#H4tJ+<2TdEX9Jbk90NcC06(-b
AU;qFB

diff --git a/docs/source/auto_examples/example_nn_Si.ipynb b/docs/source/auto_examples/example_nn_Si.ipynb
index c5ddac38..fbfde5dd 100644
--- a/docs/source/auto_examples/example_nn_Si.ipynb
+++ b/docs/source/auto_examples/example_nn_Si.ipynb
@@ -76,7 +76,7 @@
       "cell_type": "markdown",
       "metadata": {},
       "source": [
-        "In the above code, we build a NN model with an input layer, two hidden layer, and an\noutput layer. The ``descriptor`` carries the information of the input layer, so it is\nnot needed to be specified explicitly. For each hidden layer, we first do a linear\ntransformation using ``nn.Linear(size_in, size_out)`` (essentially carrying out $y\n= xW+b$, where $W$ is the weight matrix of size ``size_in`` by ``size_out``, and\n$b$ is a vector of size ``size_out``. Then we apply the hyperbolic tangent\nactivation function ``nn.Tanh()`` to the output of the Linear layer (i.e. $y$) so\nas to add the nonlinearity. We use a Linear layer for the output layer as well, but\nunlike the hidden layer, no activation function is applied here. The input size\n``size_in`` of the first hidden layer must be the size of the descriptor, which is\nobtained using ``descriptor.get_size()``. For all other layers (hidden or output), the\ninput size must be equal to the output size of the previous layer. The ``out_size`` of\nthe output layer must be 1 such that the output of the NN model gives the energy of the\natom.\n\nThe ``set_save_metadata`` function call informs where to save intermediate models during\nthe optimization (discussed below), and what the starting epoch and how often to save\nthe model.\n\n\n## Training set and calculator\n\nThe training set and the calculator are the same as explained in `tut_kim_sw`. The\nonly difference is that we need to use the\n:mod:`~kliff.calculators.CalculatorTorch()`, which is targeted for the NN model.\nAlso, its ``create()`` method takes an argument ``reuse`` to inform whether to reuse the\nfingerprints generated from the descriptor if it is present.\n\n"
+        "In the above code, we build a NN model with an input layer, two hidden layer, and an\noutput layer. The ``descriptor`` carries the information of the input layer, so it is\nnot needed to be specified explicitly. For each hidden layer, we first do a linear\ntransformation using ``nn.Linear(size_in, size_out)`` (essentially carrying out $y\n= xW+b$, where $W$ is the weight matrix of size ``size_in`` by ``size_out``, and\n$b$ is a vector of size ``size_out``. Then we apply the hyperbolic tangent\nactivation function ``nn.Tanh()`` to the output of the Linear layer (i.e. $y$) so\nas to add the nonlinearity. We use a Linear layer for the output layer as well, but\nunlike the hidden layer, no activation function is applied here. The input size\n``size_in`` of the first hidden layer must be the size of the descriptor, which is\nobtained using ``descriptor.get_size()``. For all other layers (hidden or output), the\ninput size must be equal to the output size of the previous layer. The ``out_size`` of\nthe output layer must be 1 such that the output of the NN model gives the energy of the\natom.\n\nThe ``set_save_metadata`` function call informs where to save intermediate models during\nthe optimization (discussed below), and what the starting epoch and how often to save\nthe model.\n\n\n## Training set and calculator\n\nThe training set and the calculator are the same as explained in `tut_kim_sw`. The\nonly difference is that we need to use the\n:mod:`~kliff.calculators.CalculatorTorch()`, which is targeted for the NN model.\nAlso, its ``create()`` method takes an argument ``reuse`` to inform whether to reuse the\nfingerprints generated from the descriptor if it is present.\nTo train on gpu, set ``gpu=True`` in ``Calculator``.\n\n\n"
       ]
     },
     {
@@ -87,7 +87,7 @@
       },
       "outputs": [],
       "source": [
-        "# training set\ndataset_path = download_dataset(dataset_name=\"Si_training_set\")\ndataset_path = dataset_path.joinpath(\"varying_alat\")\ntset = Dataset(dataset_path)\nconfigs = tset.get_configs()\n\n# calculator\ncalc = CalculatorTorch(model)\n_ = calc.create(configs, reuse=False)"
+        "# training set\ndataset_path = download_dataset(dataset_name=\"Si_training_set\")\ndataset_path = dataset_path.joinpath(\"varying_alat\")\ntset = Dataset(dataset_path)\nconfigs = tset.get_configs()\n\n# calculator\ncalc = CalculatorTorch(model, gpu=False)\n_ = calc.create(configs, reuse=False)"
       ]
     },
     {
diff --git a/docs/source/auto_examples/example_nn_Si.py b/docs/source/auto_examples/example_nn_Si.py
index e4e9defb..927f4246 100644
--- a/docs/source/auto_examples/example_nn_Si.py
+++ b/docs/source/auto_examples/example_nn_Si.py
@@ -102,6 +102,8 @@
 # :mod:`~kliff.calculators.CalculatorTorch()`, which is targeted for the NN model.
 # Also, its ``create()`` method takes an argument ``reuse`` to inform whether to reuse the
 # fingerprints generated from the descriptor if it is present.
+# To train on gpu, set ``gpu=True`` in ``Calculator``.
+#
 
 # training set
 dataset_path = download_dataset(dataset_name="Si_training_set")
@@ -110,7 +112,7 @@
 configs = tset.get_configs()
 
 # calculator
-calc = CalculatorTorch(model)
+calc = CalculatorTorch(model, gpu=False)
 _ = calc.create(configs, reuse=False)
 
 
diff --git a/docs/source/auto_examples/example_nn_Si.py.md5 b/docs/source/auto_examples/example_nn_Si.py.md5
index 43bb1baa..df7af95c 100644
--- a/docs/source/auto_examples/example_nn_Si.py.md5
+++ b/docs/source/auto_examples/example_nn_Si.py.md5
@@ -1 +1 @@
-61f8e82a0dc400f3c2e1e616e8f44301
\ No newline at end of file
+ddfc7cb67629dfea5b40790f4f7ce5e0
\ No newline at end of file
diff --git a/docs/source/auto_examples/example_nn_Si.rst b/docs/source/auto_examples/example_nn_Si.rst
index f8432d63..f3962d23 100644
--- a/docs/source/auto_examples/example_nn_Si.rst
+++ b/docs/source/auto_examples/example_nn_Si.rst
@@ -128,7 +128,7 @@ We can then build the NN model on top of the descriptor.
 
 
 
-.. GENERATED FROM PYTHON SOURCE LINES 77-105
+.. GENERATED FROM PYTHON SOURCE LINES 77-107
 
 In the above code, we build a NN model with an input layer, two hidden layer, and an
 output layer. The ``descriptor`` carries the information of the input layer, so it is
@@ -158,8 +158,10 @@ only difference is that we need to use the
 :mod:`~kliff.calculators.CalculatorTorch()`, which is targeted for the NN model.
 Also, its ``create()`` method takes an argument ``reuse`` to inform whether to reuse the
 fingerprints generated from the descriptor if it is present.
+To train on gpu, set ``gpu=True`` in ``Calculator``.
 
-.. GENERATED FROM PYTHON SOURCE LINES 105-117
+
+.. GENERATED FROM PYTHON SOURCE LINES 107-119
 
 .. code-block:: default
 
@@ -171,7 +173,7 @@ fingerprints generated from the descriptor if it is present.
     configs = tset.get_configs()
 
     # calculator
-    calc = CalculatorTorch(model)
+    calc = CalculatorTorch(model, gpu=False)
     _ = calc.create(configs, reuse=False)
 
 
@@ -185,21 +187,22 @@ fingerprints generated from the descriptor if it is present.
 
  .. code-block:: none
 
-    2021-08-11 22:52:40.505 | INFO     | kliff.dataset.dataset:_read:370 - 400 configurations read from /Users/mjwen/Applications/kliff/examples/Si_training_set/varying_alat
-    2021-08-11 22:52:40.505 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:103 - Start computing mean and stdev of fingerprints.
-    2021-08-11 22:53:13.620 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:120 - Finish computing mean and stdev of fingerprints.
-    2021-08-11 22:53:13.622 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:128 - Fingerprints mean and stdev saved to `fingerprints_mean_and_stdev.pkl`.
-    2021-08-11 22:53:13.622 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:163 - Pickling fingerprints to `fingerprints.pkl`
-    2021-08-11 22:53:13.662 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 0.
-    2021-08-11 22:53:13.956 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 100.
-    2021-08-11 22:53:14.244 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 200.
-    2021-08-11 22:53:14.624 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 300.
-    2021-08-11 22:53:15.100 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:218 - Pickle 400 configurations finished.
+    2021-11-20 22:33:47.584 | INFO     | kliff.dataset.dataset:_read:370 - 400 configurations read from /Users/mjwen/Applications/kliff/examples/Si_training_set/varying_alat
+    2021-11-20 22:33:47.585 | INFO     | kliff.calculators.calculator_torch:_get_device:417 - Training on cpu
+    2021-11-20 22:33:47.586 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:103 - Start computing mean and stdev of fingerprints.
+    2021-11-20 22:34:24.241 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:120 - Finish computing mean and stdev of fingerprints.
+    2021-11-20 22:34:24.244 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:128 - Fingerprints mean and stdev saved to `fingerprints_mean_and_stdev.pkl`.
+    2021-11-20 22:34:24.244 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:163 - Pickling fingerprints to `fingerprints.pkl`
+    2021-11-20 22:34:24.908 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 0.
+    2021-11-20 22:34:25.779 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 100.
+    2021-11-20 22:34:26.898 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 200.
+    2021-11-20 22:34:28.475 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:175 - Processing configuration: 300.
+    2021-11-20 22:34:29.533 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:218 - Pickle 400 configurations finished.
 
 
 
 
-.. GENERATED FROM PYTHON SOURCE LINES 118-130
+.. GENERATED FROM PYTHON SOURCE LINES 120-132
 
 Loss function
 -------------
@@ -214,7 +217,7 @@ through the training set for ``10`` epochs. The learning rate ``lr`` used here i
 ``0.001``, and typically, one may need to play with this to find an acceptable one that
 drives the loss down in a reasonable time.
 
-.. GENERATED FROM PYTHON SOURCE LINES 130-135
+.. GENERATED FROM PYTHON SOURCE LINES 132-137
 
 .. code-block:: default
 
@@ -233,7 +236,7 @@ drives the loss down in a reasonable time.
 
  .. code-block:: none
 
-    2021-08-11 22:53:15.324 | INFO     | kliff.loss:minimize:708 - Start minimization using optimization method: Adam.
+    2021-11-20 22:34:29.791 | INFO     | kliff.loss:minimize:708 - Start minimization using optimization method: Adam.
     Epoch = 0       loss = 7.3307514191e+01
     Epoch = 1       loss = 7.2090656281e+01
     Epoch = 2       loss = 7.1389846802e+01
@@ -245,18 +248,18 @@ drives the loss down in a reasonable time.
     Epoch = 8       loss = 6.7668614388e+01
     Epoch = 9       loss = 6.7058616638e+01
     Epoch = 10      loss = 6.6683934212e+01
-    2021-08-11 22:53:27.929 | INFO     | kliff.loss:minimize:763 - Finish minimization using optimization method: Adam.
+    2021-11-20 22:34:33.793 | INFO     | kliff.loss:minimize:763 - Finish minimization using optimization method: Adam.
 
 
 
 
-.. GENERATED FROM PYTHON SOURCE LINES 136-139
+.. GENERATED FROM PYTHON SOURCE LINES 138-141
 
 We can save the trained model to disk, and later can load it back if we want. We can
 also write the trained model to a KIM model such that it can be used in other simulation
 codes such as LAMMPS via the KIM API.
 
-.. GENERATED FROM PYTHON SOURCE LINES 139-146
+.. GENERATED FROM PYTHON SOURCE LINES 141-148
 
 .. code-block:: default
 
@@ -277,12 +280,12 @@ codes such as LAMMPS via the KIM API.
 
  .. code-block:: none
 
-    2021-08-11 22:53:28.005 | INFO     | kliff.models.neural_network:write_kim_model:111 - KLIFF trained model written to /Users/mjwen/Applications/kliff/examples/NeuralNetwork_KLIFF__MO_000000111111_000.
+    2021-11-20 22:34:33.901 | INFO     | kliff.models.neural_network:write_kim_model:111 - KLIFF trained model written to /Users/mjwen/Applications/kliff/examples/NeuralNetwork_KLIFF__MO_000000111111_000.
 
 
 
 
-.. GENERATED FROM PYTHON SOURCE LINES 147-152
+.. GENERATED FROM PYTHON SOURCE LINES 149-154
 
 .. note::
    Now we have trained an NN for a single specie Si. If you have multiple species in
@@ -293,7 +296,7 @@ codes such as LAMMPS via the KIM API.
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** ( 0 minutes  49.215 seconds)
+   **Total running time of the script:** ( 0 minutes  48.952 seconds)
 
 
 .. _sphx_glr_download_auto_examples_example_nn_Si.py:
diff --git a/docs/source/auto_examples/example_nn_SiC.ipynb b/docs/source/auto_examples/example_nn_SiC.ipynb
index 026f4a45..7924c047 100644
--- a/docs/source/auto_examples/example_nn_SiC.ipynb
+++ b/docs/source/auto_examples/example_nn_SiC.ipynb
@@ -26,7 +26,7 @@
       },
       "outputs": [],
       "source": [
-        "from kliff import nn\nfrom kliff.calculators.calculator_torch import CalculatorTorchSeparateSpecies\nfrom kliff.dataset import Dataset\nfrom kliff.descriptors import SymmetryFunction\nfrom kliff.loss import Loss\nfrom kliff.models import NeuralNetwork\nfrom kliff.utils import download_dataset\n\ndescriptor = SymmetryFunction(\n    cut_name=\"cos\",\n    cut_dists={\"Si-Si\": 5.0, \"C-C\": 5.0, \"Si-C\": 5.0},\n    hyperparams=\"set51\",\n    normalize=True,\n)\n\nN1 = 10\nN2 = 10\nmodel_si = NeuralNetwork(descriptor)\nmodel_si.add_layers(\n    # first hidden layer\n    nn.Linear(descriptor.get_size(), N1),\n    nn.Tanh(),\n    # second hidden layer\n    nn.Linear(N1, N2),\n    nn.Tanh(),\n    # output layer\n    nn.Linear(N2, 1),\n)\nmodel_si.set_save_metadata(prefix=\"./kliff_saved_model_si\", start=5, frequency=2)\n\n\nN1 = 10\nN2 = 10\nmodel_c = NeuralNetwork(descriptor)\nmodel_c.add_layers(\n    # first hidden layer\n    nn.Linear(descriptor.get_size(), N1),\n    nn.Tanh(),\n    # second hidden layer\n    nn.Linear(N1, N2),\n    nn.Tanh(),\n    # output layer\n    nn.Linear(N2, 1),\n)\nmodel_c.set_save_metadata(prefix=\"./kliff_saved_model_c\", start=5, frequency=2)\n\n\n# training set\ndataset_path = download_dataset(dataset_name=\"SiC_training_set\")\ntset = Dataset(dataset_path)\nconfigs = tset.get_configs()\n\n# calculator\ncalc = CalculatorTorchSeparateSpecies({\"Si\": model_si, \"C\": model_c})\n_ = calc.create(configs, reuse=False)\n\n# loss\nloss = Loss(calc, residual_data={\"forces_weight\": 0.3})\nresult = loss.minimize(method=\"Adam\", num_epochs=10, batch_size=4, lr=0.001)"
+        "from kliff import nn\nfrom kliff.calculators.calculator_torch import CalculatorTorchSeparateSpecies\nfrom kliff.dataset import Dataset\nfrom kliff.descriptors import SymmetryFunction\nfrom kliff.loss import Loss\nfrom kliff.models import NeuralNetwork\nfrom kliff.utils import download_dataset\n\ndescriptor = SymmetryFunction(\n    cut_name=\"cos\",\n    cut_dists={\"Si-Si\": 5.0, \"C-C\": 5.0, \"Si-C\": 5.0},\n    hyperparams=\"set51\",\n    normalize=True,\n)\n\nN1 = 10\nN2 = 10\nmodel_si = NeuralNetwork(descriptor)\nmodel_si.add_layers(\n    # first hidden layer\n    nn.Linear(descriptor.get_size(), N1),\n    nn.Tanh(),\n    # second hidden layer\n    nn.Linear(N1, N2),\n    nn.Tanh(),\n    # output layer\n    nn.Linear(N2, 1),\n)\nmodel_si.set_save_metadata(prefix=\"./kliff_saved_model_si\", start=5, frequency=2)\n\n\nN1 = 10\nN2 = 10\nmodel_c = NeuralNetwork(descriptor)\nmodel_c.add_layers(\n    # first hidden layer\n    nn.Linear(descriptor.get_size(), N1),\n    nn.Tanh(),\n    # second hidden layer\n    nn.Linear(N1, N2),\n    nn.Tanh(),\n    # output layer\n    nn.Linear(N2, 1),\n)\nmodel_c.set_save_metadata(prefix=\"./kliff_saved_model_c\", start=5, frequency=2)\n\n\n# training set\ndataset_path = download_dataset(dataset_name=\"SiC_training_set\")\ntset = Dataset(dataset_path)\nconfigs = tset.get_configs()\n\n# calculator\ncalc = CalculatorTorchSeparateSpecies({\"Si\": model_si, \"C\": model_c}, gpu=False)\n_ = calc.create(configs, reuse=False)\n\n# loss\nloss = Loss(calc, residual_data={\"forces_weight\": 0.3})\nresult = loss.minimize(method=\"Adam\", num_epochs=10, batch_size=4, lr=0.001)"
       ]
     },
     {
diff --git a/docs/source/auto_examples/example_nn_SiC.py b/docs/source/auto_examples/example_nn_SiC.py
index 26f047ce..26af24a1 100644
--- a/docs/source/auto_examples/example_nn_SiC.py
+++ b/docs/source/auto_examples/example_nn_SiC.py
@@ -63,7 +63,7 @@
 configs = tset.get_configs()
 
 # calculator
-calc = CalculatorTorchSeparateSpecies({"Si": model_si, "C": model_c})
+calc = CalculatorTorchSeparateSpecies({"Si": model_si, "C": model_c}, gpu=False)
 _ = calc.create(configs, reuse=False)
 
 # loss
diff --git a/docs/source/auto_examples/example_nn_SiC.py.md5 b/docs/source/auto_examples/example_nn_SiC.py.md5
index 45c470b8..993ea913 100644
--- a/docs/source/auto_examples/example_nn_SiC.py.md5
+++ b/docs/source/auto_examples/example_nn_SiC.py.md5
@@ -1 +1 @@
-19b2671ec2e63ff6e1b5732e4d7095d1
\ No newline at end of file
+dcdf008a81dc6a79cdb8b3dc7c95fcd1
\ No newline at end of file
diff --git a/docs/source/auto_examples/example_nn_SiC.rst b/docs/source/auto_examples/example_nn_SiC.rst
index c37d42be..e946413b 100644
--- a/docs/source/auto_examples/example_nn_SiC.rst
+++ b/docs/source/auto_examples/example_nn_SiC.rst
@@ -86,7 +86,7 @@ specie (take a look at :ref:`tut_nn` for Si if you haven't yet).
     configs = tset.get_configs()
 
     # calculator
-    calc = CalculatorTorchSeparateSpecies({"Si": model_si, "C": model_c})
+    calc = CalculatorTorchSeparateSpecies({"Si": model_si, "C": model_c}, gpu=False)
     _ = calc.create(configs, reuse=False)
 
     # loss
@@ -104,14 +104,6 @@ specie (take a look at :ref:`tut_nn` for Si if you haven't yet).
 
  .. code-block:: none
 
-    2021-08-03 11:20:34.072 | INFO     | kliff.dataset.dataset:_read:370 - 10 configurations read from /Users/mjwen/Applications/kliff/examples/SiC_training_set
-    2021-08-03 11:20:34.072 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:103 - Start computing mean and stdev of fingerprints.
-    2021-08-03 11:20:35.093 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:120 - Finish computing mean and stdev of fingerprints.
-    2021-08-03 11:20:35.097 | INFO     | kliff.descriptors.descriptor:generate_fingerprints:128 - Fingerprints mean and stdev saved to `fingerprints_mean_and_stdev.pkl`.
-    2021-08-03 11:20:35.097 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:163 - Pickling fingerprints to `fingerprints.pkl`
-    2021-08-03 11:20:35.125 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:174 - Processing configuration: 0.
-    2021-08-03 11:20:35.168 | INFO     | kliff.descriptors.descriptor:_dump_fingerprints:217 - Pickle 10 configurations finished.
-    2021-08-03 11:20:35.172 | INFO     | kliff.loss:minimize:708 - Start minimization using optimization method: Adam.
     Epoch = 0       loss = 5.7247632980e+01
     Epoch = 1       loss = 5.7215625763e+01
     Epoch = 2       loss = 5.7186323166e+01
@@ -123,7 +115,6 @@ specie (take a look at :ref:`tut_nn` for Si if you haven't yet).
     Epoch = 8       loss = 5.7020624161e+01
     Epoch = 9       loss = 5.6992567062e+01
     Epoch = 10      loss = 5.6973577499e+01
-    2021-08-03 11:20:35.602 | INFO     | kliff.loss:minimize:763 - Finish minimization using optimization method: Adam.
 
 
 
@@ -152,7 +143,7 @@ codes such as LAMMPS via the KIM API.
 
 .. rst-class:: sphx-glr-timing
 
-   **Total running time of the script:** ( 0 minutes  3.271 seconds)
+   **Total running time of the script:** ( 0 minutes  3.036 seconds)
 
 
 .. _sphx_glr_download_auto_examples_example_nn_SiC.py:
diff --git a/docs/source/auto_examples/example_nn_SiC_codeobj.pickle b/docs/source/auto_examples/example_nn_SiC_codeobj.pickle
index daea2e0766030f48595a58c018a0bb267829fe66..fbb4130b1f65433b53cf5852b63f4e8ada6235a6 100644
GIT binary patch
literal 10218
zcmds7-;3Qu6yC0Pckk}qwYyzgckN=qVyz01iXiwCbS;W?t?q(QC?U<A+a#M!mL#`z
zrGk$x=1FhCM_>9b3WDe#pfA4o=!4*cAPR!df}Y7Fb7qpc$=pzt+L!hw=lkY6XU>^9
zGwFLXH-CI+D*Kc6?9lg#ow#8TkJ8@jQJ+Mlzv<e^=%dlYZ;sBUvk{4helof|dKZ3m
z0&9<q((YCmke!2l_~-0i*dO}z*_EN|Cobh#WY5gl38Q4>ETmmGHf`UE<I&;JO6NAM
z#EJ>!JPV}G96oX8)8o6oI~W*!W>fq<a(d~K^k(ENzCWzO%SM{8nT#7oan@V&9u7&`
zxe$Vv+%fhOyjoz|efkz$k$!SNxHYOdgKGF&30&r)BBr+<XY&Q2y~_<?3g(6B0UsZ8
z7U@JBgLxvpX&Y9*Z~E2&i3I6uTVxnn{uW8DhtaNR0&FLlI~hJUg3Qtkm?`ydQG5?G
zb(qgoEolkVk_BV;432yJHR|b37}-wAlQmJP!2*&{JDLm6>oAmMJ1^*<Q07Z$z5+6#
zm`W1w&GZ;SK;^;fIRBd}e7Om{Tn1Crt4*kGfuh!1P1Z9!U~f=UlCJhiY)9@sJU6bA
z#Eji{2o}Zq_QBpBNuq=E!ysFF&<0ZL5jG*n5F2rBZw?9oJTV6Wh?~ru=V85s(;5<J
zU@l=`>IqjWgjBlfsis2A9t$<BZYrV<5#6YaZeVm9g^XiWsc-OU92I)wv;0{Q^zpTH
zxtun1S^HPAMLDu=k_>TzlsqlssN}&Q-e5uOB)4OaEmm&6X+zL8_E;4R6&^B$J=^ka
z!;XlR2+PBRFT&HIZ^7cRLl@$0vTsFTvAs`h7Z$(V9&|jw5}UFxdw9wJZ5E4?o*6GZ
z{o(VO!y)O`?tF;ztxtuaxry)-r9e#Xxk@2-w1!@V<FzgMtHXxQeZuB_dh1+$e{zf3
zS!tJT$FYlh$CA{Q&=sI<D^g`DZ{%aUm?{)q{EWHS$vQp0cAcPYp(Gu4dg4?44c#2;
zaBJ{5^jB=?k~*%T^t?srihjGw)EK()P}Kj8!#m-HEIe8H(-_q?X_Z+jhAMGrODy3~
z-}k6*7WC0m4`{3D&QhU`;;so$Rd3dW5h#02w%C{bgwk6~9&PEC`{DJ#53RnLpUki(
zyG2_@!-2;@SPOrp8ObuZiU;E!Tfm2j>u1Lz`8Qxmy3W=B+knl?jIC?Lgx1lb8c@I~
zsj?K7FjU)RLWjFCwTd9<kYT?4iP77@dQiJJQH2z88d<d+Yh>bHd2>gmrqHo9R!C~6
zYD&vn(5F$AnYW~m7fa4t*2nSk^Hy^Zlu@OL<?D0x6{PJy+TRu{rE+A={?^;be5<9J
zpvA4_!6Z5@4Z45_AG!p!JMR=Dm&jUZFEOBMLsG4b`UJz#)$#sPi&*NFafxg)gs|P#
zZn}u3#<<F6(|cNql?Dxl`8f%Cv=kzmA)2nLl<|FB#j660>6?oHsc6uZS0q<7qM+%S
zmRYv{F4I$Fc0pr@!N9#L+z_tj_qftqi0%Y8xW!I1j%tFF<~Rxr=A|r{v}`)`4D;U=
zN8z=`d@3`_vvW^aYo$ShK^X8L95EpRi~;ZSVfw@l&fCq+;np7RA9t+4!LW6*pMW@-
z9ks;^N(=7wu37Z60=-`nHc(zZ9IV=%>k$H712^u#H!?qpq7!=XnjS-`LV#*37ch<5
z1zS}Afux>?uFKlzfCU+UWQDs#PoT!08@PMY?dvJ^Q4}p<LbcAMIz*UvAM?DUUx%`<
z4z!!cY&uqI;^|A8Tg<HtGs%~RF?8A;?HD{dbSrm=RnSqfLC>YP>d93^+M-opRQ#!=
zxL0)Ex!RgS@IJ3as_MY|q81{Sf%j#tDOSPPxgAO8@)W}Csh(=vCQg+}JN#8B2LD?w
zelb5loBLKmS5uZi^B2R|efuJP1X({v-}&wpSUtV(a-@~s$J~xx#c}C!@hw&7jM+9e
zI^5dP(jmk6{U9Yt<9V}H?KQG$`_{;Wj{Ph-*7`jBi$0CAUVqcaOU3)UJ`OASj~oPF
z3VBNZFTX*1e@WqK_Zixoo))i6#*>uO)0@!*Vm5Xq#KhmDy`BbMnSXa{0HwrR(Ewn^
z-IH6x-ht;Bw(@p-H?I5_32!|ISm_c^(L1|gIU@y3)he`FG*v|9M6BG=7bhSU7?bgw
zq<E5s<%wFPG6A2gg-EIQRIMo{-LttJ++t_Vcy@gq?k8DC9+`0>efNBHOk=J@={D(;
zl49`BbMf<V^1I*$#EP)rn#~`8j%wrRGvhHQKLM(m2=%E8hZAR_$rh*Lt<0M{_53%G
T)eFJELmwQ4@drHt4voJ7Gs*gs

literal 10227
zcmd^FL5ST{6rI-T{6F*O(`kp+87&s7GHunMVmAdhI_SbWmUci8N=WnHG)XhbnB=#P
zRB)%26oh_3Pz1q^3%4!>Q4j>7;L2T>iXaMt3pd`Em%RI4{!8+bR-xF<^e6Y8ckg}Y
zzI)&O|LuJr{Jb`k{YwXS==;P@+%SkI>EM-UL?SX;ckN_yb8_O`<aj#t!#Fm!-N4;;
z-z1X@ljCo})7ik<CX@8&8F;xt#*yW3kmPC@U4A(n5g*0Vx$ST?_Ca*vvhQwf85C`j
zF0!X)?1WKLde?Vj)AlV8yFP}coL=)P&V0HrgP%A9XUQ2}9~1D*@@>P8h?Nvp1Jt9>
z!_%>EC1JD~Mz(XF>{t;<oZlg~OX5Nv0FN<yDGHN^m;C=G=-UpgrY8mlF;DcDKQOgC
z^=*~<))opwa|G27mjcn=bCreMx*B+u-XK*y)j>n&*4V!H>@e7JFU7!A7N}X@jgtxN
zX#e8a^%EC}4v~DEN?Yk_#_rg<Ld<X{$tWI~aUyZ%u?mtawO}2Q>A(=_BX_ASDEfJd
z`MJmB9`-lFfNbvWkSQ-uGcS8VU~F1}0~$1YArEG=k6`SB8IuhVU?~|J7MuoGEP5hP
ze)o&_6KD8tPCc%?RJ<&d8#M&D2JC%Bia9Ug%5!kM(9;AAzrVb8ZbOv`f~vOJf=;91
zf-IW;8k)|7)@PX=u<+sspaw~M>sDgLB*C5HPr&CxOnSs*jQlV4cqqLQ_DWPQ3u&q$
z^)913Epe%DtH3bgo3;V|nOX8|lf<G<ptnb*Vxo|wlx4nVXt5Dw)yE82`Ju5`6yHNA
ziW5@@yQF((U>LY-XvGnrF!48O{ZPBNBAQQ04KXWnq0+ZTRBgu^k<hDm(5n^j3KCQ8
zQBA5k?`dMFa@M(_sVpY1^S-7MX7Np`LDJc*ci~6qVRc!cXOnc~S5<%-eGKu(`O?YV
zI1`FQXq~&PW{7HN;pfc4J|k-EVlEG*T&8j9nz_?R+H{^)Mbp1zruUfX_B~|zH_Y-S
zN(<u>9H-dy<OO~?FC^02b|s{sOBIwxIcb8xy|~3ZS{RYoj@%tM@Nhep&hKtRsfc!8
z7zf!^g>va+Uh%kmY{a>kxs}V&C+1dGXj_T%lA&DrVU3lc4mm$jmssj3aq^ZIJ*JIm
z(FjXV>v1rMw^<OEQ90ZG7%y(n`mL6ui8B9HSnEyDjqfiQ-+I6ffvyL{i2sf@EU$Aa
zN)bIWO&8ps#vQLh?ch@t6ngVF<ImoWry%(lbMJG5smqf$vu9?8idyL`y%S@mK2*9@
zmaf+{Z~v8hwZ^)t3+BL^&plzzYQXSdd5gI!Nry%4GiX<f(Ksj-K=u+M(6tIyz-gDl
z3VrlwRSL#_Xz6&PO8o-UJN2e2AZ?>n%4qQ;lEnkCIBP<4Yis&WZ>?5S)opqwYgNP^
z(|e*;ly{cilerv87xT-Uxnm{tp3Vq4e>RNWH_p;W5d9u~=erl-(C|*>npQd&k$CLW
z8za-HoCbKOxlXAox=@S-O3yoxwU`l1BQ<V$wJL7-i<t|B<N{K(H9IRXw5=?yjV7Lx
zOccffHFQ*tSp-!TADeVULN>`FBQ44L4!cNiN=@e|=UbB?5b&<ni(=C?)>YSwKnjlr
z4D#Z2Np*jzQno0KD$Dq6PKB#d;Eg6N;Q4@C7#pcz(3NYFD;iO_tD8s}_5haL|JyLh
zyOBk|xLP#MYdY2JM%2c0QrkYH)rXQ}T}MjZNBS_TbJhDqUtc_9y-)S^@VxyjSAy{p
zohM=Q$UedWSUXHA4m2)}(flGqo)V~qqp7vbQ_C8SOyAlikzm$}+|vpo9h!xLBekH~
zOshq~xO^pT(;;jj4B45wyz#!)1SuQueXFU6_I@X68~}T1k1N|+(^`1nYv?QU;0Fyw
zDFJS2C?NBH%%!3CWsHL_2JY$W({0Ip)y1WbQkA296;PR0t(1{kze*Ojxjy|?tEr^Z
zAGInHk^Za|MGF0u%aQb8K_MY8c$ZmwFm%0**Me&APb~^=)a+h1aA7yRj%QX+6Qpds
zx1gzr_6{YD_Gi|ThQ5+y%NmLj)edVYAmLVXX_8J)bUEFoUqNYXHGdwo6X(spa%IhN
zs&_2&t>_n&Q29jy$CuR;>slNn$_Z)F1stK$`%t&ZX|Zw%tb^?(1ho1Y$?7`|6>Ylq
zZamEgILeXhvpMsubZHb`4gAm=nZ@`2oO%7?iw_z-@#ikybGbS@K&Bx;e!%9`<HYr|
zZIwU6w3npw@H3G!Za^&I%bB_S0U%6wAwCnX_#KI$s!C?{&;pke^SDVCQ}|VB?}Qo4
ywDnL^vLBnMf2ks<>RZpcw`_Zg->=ZG00dWfG7I=<{g!OK5Dd0^%3qa?jeh_qn*G55

diff --git a/docs/source/auto_examples/example_nn_Si_codeobj.pickle b/docs/source/auto_examples/example_nn_Si_codeobj.pickle
index 7a354061cbd4aadeee207c4f28d3f11b2357d018..bee18e68bba9e8ec497700d77a855c8a2003fc5b 100644
GIT binary patch
literal 8388
zcmdT}&1)n@6whWSUo#W)MY4&z5>b%#gO>H=O>lz;S!2|IC++Q>sbsp-(;d2d5+jNq
z2Q|>(!SN`9Ac$VPc<|uOgCGimC&7bv@jqDbRaJMrs_vTZn#p=_FPZLo?^nO~S+8r~
zo%!_p2dCJ-wABfHpL7y84B}zhdNJyfh;(<{PBQ#t_~^^S>*;Jn;(?zGUmLy+PmRFo
zlVLi)8wO<m>==H|_QUSLr|<3zTt9IsPMg28<6amg!(KC;cVoNbJ8?WbA2{jUj*~bs
zp_tbJv^R%udJF0Lk?$THSY0lY{~h*P>7ueS?6p4{l+}wuhH5hm8+Pe!O^ka!AZg=P
z2v!Qi_#;@=<lKF_3Z^KJ!VV#gY#u>2VpUNtVo?&)yHD|Wfom_j0fb;d3J=)$WUoyF
zvDPJVCvuOIFtQFwV#n^A1i5d$a@OyYBszO;5U>nE8Nhoi3x+_5tvHjn4{`-~V;=<6
z;IYJk91m<Yg@+9&(1<=wtu+B$DFIUIy0?aa7{_Oj!`hDUnc~rFrQI!zPc4yAs5<sF
zHjRQpYo4dpkaPy>ocNeAi@RhHIsPt5PQ&PkX5Ce86O)UrfGyiWzTQgP@;X#>9;Y%=
zL2?{b<ozn=-IV+)N_DxI?v(#nX*+3_Yj-wH?%U`zEzSh-*W#cP54jT?$-QXddBDY)
z08Te(;W{;Bm~e)zj^lT%PDGqUYHJ%e;qAb8py}<0QKy#+z{U^pY@}*jAhz(g4GNu}
zREM|(EnQ@81$-6?MNgEX^40_m&3%uXyPh4H_OX-nEH88e`U9(TbuWzFH};f8FnW$k
z`R*OsDL+6v7ApnMQE+2#5RrQ56g5_uMiq~bxkb~vH-J9x+}d-D`Q1qy>{u0*yg|=k
zr4_xv`V<)3<Pp{%!+BCw=%dp=;}i>xN^mH=hpoqLh52w(MyhRvqg9dr*Em=-nR{3<
z(4My`_hgH;Zp{6{-ePnqU=+m4-8!dO&_jOTA-`GRM@LNHc7+w86$F$Qyo<BHv@GZI
zHOPt_UQ%7omvdDE9SO1)e}>`HD*6%kpCEYjU|sHpr-2_jT|293tm656xT5aBnH#qK
z1uLwq55g@T*Foa?Yz9@I*@0f}{34;tO91(w9*=npwL6<0kAGk+v^ZlNkCl8dasR>{
zn>gvwXiYZY<=Ym(CLmrwK+1n3<%>FrWiRU)mecLpzH>$*>3e8XY0I(*_-EV>nfL2T
z(ArL9go3`hi&eG`eTn${Y>1GI1xu=0s5f<KYRUwp$lSYZs3^#Kvx>$F@1q5;SF}`b
z!9ZWvSG~4@qTF1)B?AS~VZGH%8agGNLfL0jCkzhULn+tRvTFuq6@oLy1w)=2qoBr^
zR7XKzu(pL!4LS^B%U`-F22z%6w3pnAQB{XK!HlYdL93n+R$a2iFo*X?V2XT*c{eg~
zxFv!^e%}eC0k**&LE~!iD7PHApg3{pHP)z8T`24FRXaNf4LMP(B{4&vWp?DfTN*<u
z%1w(csTwnvbZWUvvdI1=g*`fgPY?ljGCr``VUZ5iR^JWWzH*JUt)IqG5&~*8mYV=!
z-n}mJZk-=Tr;(cw`^fFv>>11J0rgBO%F%pNCSvnjjC$l^j~{s<QV)fuA7!TV52yR3
zQ%Yl7)loH)O%0~8RW&$h(urcyMs|>C%&90Z@5~fLS7_c_rrL7JdGDBNVe!46sgSfS
zf6<HPxgb8a=|LcUzIz{LnofF!=RBN^cI=!G8}4x(3Wiy~t!8~$@&b8pf^D25CSaH-
zL&e|Mq}<n9I8pZ3q8i&)i-d-Kt{AqNRe;v0sv7vh97bCOUz+PHb?}wB9#+CPnG#9o
zaJu7Nt>L?I>HjsnZ!-laZDs`k$&Wd|TF802c!4w@0cRZj5&9VO9~7r8$;q5ipD2HI
zK*pqX%E<U9g>fs-(u{3YV*XsIsm;Y-D^-;A`>j$GGw%0Hj-*>`oMFu~JWkk56WMX1
zTqQp~Vzkj}L82LrIpcz&^Zyj&7hrb2<p#uwF!^V*7r>(&2cHX%x<x}(T_E&xE9%IR
zsFTG|{KKsIzfRD1RrZoIaBPMvMSNg0bm>z{T)&l8AdDCJ2F(4c!yV%runGnf^S)_%
P<549$#+GoyG_d{!h1pWs

delta 1538
zcmZ`(U1%It6z*hZ@9fO(WH;OW|H~%rHoIFJEisL49?Ggso0{TNMWt!F#tk=<%|tT0
zN(z<uBDIX*dPkx~rLC|}whgo|LZPL!f~b)|!H52UsKp0~?Sl_Osh)f9?(7Qsa`?XQ
z{M~!cnYmH?z9=obKDu6)3I)Y51}A2Xa%s_?v~wryoFy5h`Lb=HrfQ-2h1YA9vQ30^
zJZpJ5n=<jjV%4_X)rG}rJZKmIXU(bUO35^BQb-R88abNY4)#<!Dd8yA$ocdhpuaI!
zvX{CJ^nz;uTyu>6irx^l_iE%X)JK-&LJRv{bCubd85?`>GZ_43(g^H_?2W)qu$j$y
zum?BCB|QHBO<o39KA4wS`nsE@$CMpn8RUrf17HulDbjG|L<5uuUg`%$Yj>`J{!ojM
zC2bgd6n(7~80r(3ot!c0Fgx{>5+!eYG7f&EsA3(W9!sYnXWhknod#vyMSFqL(d#Ve
zzXxlepAQv?70NI)B;gC-7!41Q?(h>JjfQC|@Q&24AmZ=v7IHV7knj~oMkIW-sUC}X
zC48Neu!~KOzUoc(s#dJBV49?q3X@{$FjQVi6(zg|(c{XbsGY14=T=QjZ*?2|htvqk
zsPssO)DdPQ<Wp4@?}P6@hmP={xT7GX+!M0sF+E0!ycB!|gw^0KBnz81L5bD)cE!se
zbsrIi`ym$=70Z@10Y|p6VmX7-L(L>%2UNKwkI8zPD`|ID*84c#?;e!(tsF18^Qu0;
zFx@Ss>k~-aCvGF(bPdS*Q#^?B4UrCfpZKJ#k8pKO%#e>ox)5vPL1r^#PE=I=S*P4D
z@gv|UCia4TCy_@l^32u50T5ObNfJuZ@d9eEx&A5-`XL!5$CF7k*`}A1A#|)ovuL^n
zqkd<c`V1F8?|BUh3xP3aJIk9qV5N7SjrC%0oTLKqxxrWg2ikAB%mTSgL!i=R_Z(QC
zFH1JM>A`(Ic*#01SoG~hb_@1HfM3>E9Nuxi(_DP$+k-yiaHoF=ooT6Qa?-D%FC5x&
zw+0uVL*vl<eY7hFE5lboe!f&KndR!gx8V|tr*r)|@+SQpKa3X0yHWOKQut*|Mb~)E
z-LX9J#^@IBj?wCrQ}?&reI-^zYb}^1xp)Yzw<(Tm=vJGaiL2;GP8;s6s(y!Iy1s7(
zwkbb3?3`y<*6;Jc+n!PA+K4=j{$N7=!~}3b2-ffQGmw(?M?92^Uive$&|=xFRjTB8
iV2k4Fq8|)x^PT@F<QP2}@<I>IHhK!4K+{{U4gLcKrnMdb

diff --git a/docs/source/auto_examples/sg_execution_times.rst b/docs/source/auto_examples/sg_execution_times.rst
index 0baa6014..4de9a2bf 100644
--- a/docs/source/auto_examples/sg_execution_times.rst
+++ b/docs/source/auto_examples/sg_execution_times.rst
@@ -5,16 +5,16 @@
 
 Computation times
 =================
-**00:22.986** total execution time for **auto_examples** files:
+**00:51.988** total execution time for **auto_examples** files:
 
 +-----------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_example_linear_regression.py` (``example_linear_regression.py``) | 00:22.986 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_example_nn_Si.py` (``example_nn_Si.py``)                         | 00:48.952 | 0.0 MB |
++-----------------------------------------------------------------------------------------------+-----------+--------+
+| :ref:`sphx_glr_auto_examples_example_nn_SiC.py` (``example_nn_SiC.py``)                       | 00:03.036 | 0.0 MB |
 +-----------------------------------------------------------------------------------------------+-----------+--------+
 | :ref:`sphx_glr_auto_examples_example_kim_SW_Si.py` (``example_kim_SW_Si.py``)                 | 00:00.000 | 0.0 MB |
 +-----------------------------------------------------------------------------------------------+-----------+--------+
 | :ref:`sphx_glr_auto_examples_example_lennard_jones.py` (``example_lennard_jones.py``)         | 00:00.000 | 0.0 MB |
 +-----------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_example_nn_Si.py` (``example_nn_Si.py``)                         | 00:00.000 | 0.0 MB |
-+-----------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_example_nn_SiC.py` (``example_nn_SiC.py``)                       | 00:00.000 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_example_linear_regression.py` (``example_linear_regression.py``) | 00:00.000 | 0.0 MB |
 +-----------------------------------------------------------------------------------------------+-----------+--------+
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 936e3d36..cb8d01a3 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -39,7 +39,7 @@
 # The short X.Y version
 version = "0.3"
 # The full version, including alpha/beta/rc tags
-release = "0.3.0"
+release = "0.3.1"
 
 
 # -- General configuration ---------------------------------------------------