updated readme

Author: Frederik Rahbaek Warburg

README.md

@@ -125,7 +125,7 @@ If you want to cite the framework feel free to use this (but only if you loved i
 ```bibtex
 @article{software:stochman,
   title={StochMan},
-  author={Nicki S. Detlefsen and Alison Pouplin and Cilie W. Feldager and Cong Geng and Dimitris Kalatzis and Helene Hauschultz and Miguel González Duque and Søren Hauberg},
+  author={Nicki S. Detlefsen and Alison Pouplin and Cilie W. Feldager and Cong Geng and Dimitris Kalatzis and Helene Hauschultz and Miguel González Duque and Frederik Warburg and Marco Miani and Søren Hauberg},
   journal={GitHub. Note: https://github.com/MachineLearningLifeScience/stochman/},
   year={2021}
 }

stochman/nnj.py

@@ -712,7 +712,7 @@ def _jacobian_wrt_weight_sandwich_diag_to_diag(self, x: Tensor, val: Tensor, tmp
             output_tmp[i, :] = output_tmp_single_batch
 
         if self.bias is not None:
-            bias_term = tmp_diag.reshape(b, c2, h2*w2)
+            bias_term = tmp_diag.reshape(b, c2, h2 * w2)
             bias_term = torch.sum(bias_term, 2)
             output_tmp = torch.cat([output_tmp, bias_term], dim=1)
 
@@ -1194,4 +1194,4 @@ def forward(self, x: Tensor) -> Tensor:
 
     def _jacobian(self, x: Tensor, val: Tensor) -> Tensor:
         jac = 0.5 / val
-        return jac
+        return jac