setup.py

from setuptools import find_packages, setup


# Read contents of readme.md to use as long description later
with open('README.md', 'r', encoding = 'utf-8') as f:
    readme = f.read()


setup(
    name = 'deepfusion',
    author = 'Atharva Aalok',
    author_email = 'atharvaaalok@gmail.com',
    version = '0.0.3',
    url = 'https://github.com/atharvaaalok/deepfusion',
    license = 'MIT',
    description = 'DeepFusion is a highly modular and customizable deep learning framework!',
    long_description = readme,
    long_description_content_type = 'text/markdown',
    packages = find_packages(),
    install_requires = [
        'numpy==1.26.4',
        'matplotlib==3.8.4',
        'setuptools==69.5.1',
        'typing_extensions==4.11.0'
    ],
    extras_require = {
        'visualization': ['graphviz==0.20.3'],
        'gpu': ['cupy_cuda12x==13.1.0']
    },
    keywords = ['deepfusion', 'deep learning', 'neural networks', 'artificial intelligence',
                'machine learning', 'model', 'optimization', 'backpropagation'],
)