tinygrad/setup.py

34 lines
834 B
Python

#!/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="0.3.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.8",
extras_require={
"gpu": ["pyopencl", "six"],
"testing": [
"pytest",
"torch",
"tqdm",
],
},
include_package_data=True,
)