-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
25 lines (24 loc) · 841 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from setuptools import setup, find_packages
setup(
name="cudatiger",
version="0.0.1",
description="About An accelerated implementation of the Tiger optimizer for PyTorch, supercharged with Triton for enhanced CUDA GPU efficiency",
long_description=open("README.md", "r").read(),
long_description_content_type="text/markdown",
packages=find_packages(where="src"),
package_dir={"": "src"},
author="juvi21",
author_email="[email protected]",
url="https://github.com/juvi21/tritonizedTiger",
install_requires=[
"torch",
"triton"
],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Environment :: GPU :: NVIDIA CUDA",
],
python_requires='>=3.10.12',
)