Merge branch 'master' of github.com:lab-ml/nn

merge

Makefile

@@ -22,11 +22,11 @@ uninstall: ## Uninstall
 	pip uninstall labml_nn
 
 docs: ## Render annotated HTML
-	python ../../pylit/pylit.py --remove_empty_sections --title_md -s ../../pylit/pylit_docs.css -t ../../pylit/template_docs.html -d html -w labml_nn
+	pylit --remove_empty_sections --title_md -t ../../pylit/templates/nn -d html -w labml_nn
 
 pages: ## Copy to lab-ml site
-	@cd ../lab-ml.github.io; git pull
-	cp -r html/* ../lab-ml.github.io/
+	@cd ../pages; git pull
+	cp -r html/* ../pages/
 
 help: ## Show this help.
 	@fgrep -h "##" $(MAKEFILE_LIST) | fgrep -v fgrep | sed -e 's/\\$$//' | sed -e 's/##//'

labml_nn/__init__.py

@@ -16,7 +16,7 @@ We will keep adding to this.
 contains implementations for
 [multi-headed attention](http://lab-ml.com/labml_nn/transformers/mha.html)
 and
-[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html>).
+[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html).
 
 #### ✨ [Recurrent Highway Networks](http://lab-ml.com/labml_nn/recurrent_highway_networks)
 

readme.md

@@ -15,7 +15,7 @@ We will keep adding to this.
 contains implementations for
 [multi-headed attention](http://lab-ml.com/labml_nn/transformers/mha.html)
 and
-[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html>).
+[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html).
 
 #### ✨ [Recurrent Highway Networks](http://lab-ml.com/labml_nn/recurrent_highway_networks)
 

requirements.txt

 torch>=1.6
-labml>=0.4.37
+labml>=0.4.52
 labml-helpers
 torchvision
 numpy>=1.16.3

setup.py

@@ -5,7 +5,7 @@ with open("readme.md", "r") as f:
 
 setuptools.setup(
     name='labml_nn',
-    version='0.4.10',
+    version='0.4.12',
     author="Varuna Jayasiri, Nipun Wijerathne",
     author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
     description="A collection of PyTorch implementations of neural network architectures and layers.",
@@ -17,7 +17,7 @@ setuptools.setup(
     },
     packages=setuptools.find_packages(exclude=('test',
                                                'test.*')),
-    install_requires=['labml>=0.4.41',
+    install_requires=['labml>=0.4.52',
                       'labml_helpers',
                       'torch',
                       'einops',