Merge pull request #18 from goktug97/master

add setup.py and change imports to relative
This commit is contained in:
George Hotz
2020-10-26 09:09:31 -07:00
committed by GitHub
5 changed files with 33 additions and 3 deletions

3
.gitignore vendored
View File

@@ -1,3 +1,6 @@
__pycache__
notebooks
.*.swp
build
dist
*.egg-info

24
setup.py Normal file
View File

@@ -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)

View File

@@ -0,0 +1,3 @@
import tinygrad.optim
import tinygrad.tensor
import tinygrad.utils

View File

@@ -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)

View File

@@ -1,5 +1,5 @@
import numpy as np
from tinygrad.tensor import Function, register
from .tensor import Function, register
# ************* basic ops *************