mirror of
https://github.com/labmlai/annotated_deep_learning_paper_implementations.git
synced 2025-08-06 15:22:21 +08:00
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
import setuptools
|
|
|
|
with open("readme.md", "r") as f:
|
|
long_description = f.read()
|
|
|
|
setuptools.setup(
|
|
name='labml-nn',
|
|
version='0.4.99',
|
|
author="Varuna Jayasiri, Nipun Wijerathne",
|
|
author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
|
|
description="A collection of PyTorch implementations of neural network architectures and layers.",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/lab-ml/nn",
|
|
project_urls={
|
|
'Documentation': 'https://lab-ml.com/'
|
|
},
|
|
packages=setuptools.find_packages(exclude=('labml', 'labml.*',
|
|
'labml_samples', 'labml_samples.*',
|
|
'labml_helpers', 'labml_helpers.*',
|
|
'test',
|
|
'test.*')),
|
|
install_requires=['labml>=0.4.110',
|
|
'labml-helpers>=0.4.77',
|
|
'torch',
|
|
'einops',
|
|
'numpy'],
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
'Intended Audience :: Developers',
|
|
'Intended Audience :: Science/Research',
|
|
'Topic :: Scientific/Engineering',
|
|
'Topic :: Scientific/Engineering :: Mathematics',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Topic :: Software Development',
|
|
'Topic :: Software Development :: Libraries',
|
|
'Topic :: Software Development :: Libraries :: Python Modules',
|
|
],
|
|
keywords='machine learning',
|
|
)
|