diff --git a/.gitignore b/.gitignore index a783405e0c..c42d6ff5a5 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,6 @@ __pycache__ notebooks .*.swp +build +dist +*.egg-info diff --git a/setup.py b/setup.py new file mode 100644 index 0000000000..fc54c279aa --- /dev/null +++ b/setup.py @@ -0,0 +1,24 @@ +#!/usr/bin/env python3 + +import os +from setuptools import setup + +directory = os.path.abspath(os.path.dirname(__file__)) +with open(os.path.join(directory, 'README.md'), encoding='utf-8') as f: + long_description = f.read() + +setup(name='tinygrad', + version='1.0.0', + description='You like pytorch? You like micrograd? You love tinygrad! heart', + author='George Hotz', + license='MIT', + long_description=long_description, + long_description_content_type='text/markdown', + packages = ['tinygrad'], + classifiers=[ + "Programming Language :: Python :: 3", + "License :: OSI Approved :: MIT License" + ], + install_requires=['numpy', 'requests'], + python_requires='>=3.6', + include_package_data=True) diff --git a/tinygrad/__init__.py b/tinygrad/__init__.py index e69de29bb2..e218b10dfb 100644 --- a/tinygrad/__init__.py +++ b/tinygrad/__init__.py @@ -0,0 +1,3 @@ +import tinygrad.optim +import tinygrad.tensor +import tinygrad.utils diff --git a/tinygrad/gradcheck.py b/tinygrad/gradcheck.py index b5f4c8093e..866fdd475b 100644 --- a/tinygrad/gradcheck.py +++ b/tinygrad/gradcheck.py @@ -1,7 +1,7 @@ import numpy as np -from tinygrad.utils import mask_like -from tinygrad.tensor import Tensor +from .utils import mask_like +from .tensor import Tensor def jacobian(func, input): output = func(input) diff --git a/tinygrad/ops.py b/tinygrad/ops.py index 92f5d78fa1..1ef53358e3 100644 --- a/tinygrad/ops.py +++ b/tinygrad/ops.py @@ -1,5 +1,5 @@ import numpy as np -from tinygrad.tensor import Function, register +from .tensor import Function, register # ************* basic ops *************