Add scaffolding

mcbal · Sep 4, 2021 · 9c5bb97 · 9c5bb97
1 parent 2f6632e
commit 9c5bb97
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -127,3 +127,6 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# VSCode settings
+.vscode
diff --git a/README.md b/README.md
@@ -1 +1,9 @@
-# Approximate free-energy minimization
+# Approximate Free-Energy Minimization
+
+Experimental implementation of approximate free-energy minimization in PyTorch.
+
+**Summary:**
+
+**Blog post:**
+
+## References
diff --git a/afem/__init__.py b/afem/__init__.py
diff --git a/setup.py b/setup.py
@@ -0,0 +1,14 @@
+from setuptools import find_packages, setup
+
+setup(
+    name='afem',
+    packages=find_packages(exclude=['images']),
+    version='0.0.0',
+    license='MIT',
+    description='Approximate Free-Energy Minimization',
+    author='Matthias Bal',
+    author_email='[email protected]',
+    url='https://github.com/mcbal/afem',
+    keywords=['artificial intelligence', 'attention mechanism'],
+    install_requires=['einops>=0.3', 'numpy>=1.19', 'torch>=1.8'],
+)
diff --git a/tests/__init__.py b/tests/__init__.py
-Original file line number
+Diff line change
@@ Expand Up / @@ -127,3 +127,6 @@ dmypy.json @@
     # Pyre type checker
     .pyre/
+    # VSCode settings
+    .vscode