mirror of
https://github.com/labmlai/annotated_deep_learning_paper_implementations.git
synced 2025-08-06 15:22:21 +08:00
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
import setuptools
|
|
|
|
with open("readme.md", "r") as f:
|
|
long_description = f.read()
|
|
|
|
setuptools.setup(
|
|
name='labml-nn',
|
|
version='0.4.128',
|
|
author="Varuna Jayasiri, Nipun Wijerathne",
|
|
author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
|
|
description="🧑🏫 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit), optimizers (adam, radam, adabelief), gans(dcgan, cyclegan, stylegan2), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, diffusion, etc. 🧠",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/labmlai/annotated_deep_learning_paper_implementations",
|
|
project_urls={
|
|
'Documentation': 'https://nn.labml.ai'
|
|
},
|
|
packages=setuptools.find_packages(exclude=('labml', 'labml.*',
|
|
'labml_samples', 'labml_samples.*',
|
|
'labml_helpers', 'labml_helpers.*',
|
|
'test',
|
|
'test.*')),
|
|
install_requires=['labml>=0.4.153',
|
|
'labml-helpers>=0.4.88',
|
|
'torch',
|
|
'torchtext',
|
|
'torchvision',
|
|
'einops',
|
|
'numpy',
|
|
'fairscale'],
|
|
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',
|
|
)
|