Update docstrings

Author: ziatdinovmax

gpax/models/vi_mtdkl.py

@@ -1,3 +1,12 @@
+"""
+vi_mtdkl.py
+========
+
+Variational inference-based implementation of multi-task deep kernel learning
+
+Created by Maxim Ziatdinov (email: [email protected])
+"""
+
 from functools import partial
 from typing import Callable, Dict, Optional, Tuple
 
@@ -15,7 +24,7 @@
 
 class viMTDKL(viDKL):
     """
-    Implementation of the variational infernece-based deep kernel learning
+    Implementation of the variational inference-based deep kernel learning
     for multi-task/fidelity problems
 
     Args: