Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
labmlai
GitHub Repository: labmlai/annotated_deep_learning_paper_implementations
 Path: blob/master/setup.py
1689 views
1
import setuptools

2


3
with open("readme.md", "r", encoding="utf-8") as f:

4
    long_description = f.read()

5


6
setuptools.setup(

7
    name='labml_nn',

8
    version='0.5.1',

9
    author="Varuna Jayasiri, Nipun Wijerathne",

10
    author_email="[email protected], [email protected]",

11
    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. 🧠",

12
    long_description=long_description,

13
    long_description_content_type="text/markdown",

14
    url="https://github.com/labmlai/annotated_deep_learning_paper_implementations",

15
    project_urls={

16
        'Documentation': 'https://nn.labml.ai'

17
    },

18
    packages=setuptools.find_packages(exclude=('labml', 'labml.*',

19
                                               'labml_samples', 'labml_samples.*',

20
                                               'labml_helpers', 'labml_helpers.*',

21
                                               'test',

22
                                               'test.*')),

23
    install_requires=['labml',

24
                      'torch',

25
                      'torchtext',

26
                      'torchvision',

27
                      'einops',

28
                      'numpy',

29
                      'fairscale'],

30
    classifiers=[

31
        "Programming Language :: Python :: 3",

32
        "License :: OSI Approved :: MIT License",

33
        'Intended Audience :: Developers',

34
        'Intended Audience :: Science/Research',

35
        'Topic :: Scientific/Engineering',

36
        'Topic :: Scientific/Engineering :: Mathematics',

37
        'Topic :: Scientific/Engineering :: Artificial Intelligence',

38
        'Topic :: Software Development',

39
        'Topic :: Software Development :: Libraries',

40
        'Topic :: Software Development :: Libraries :: Python Modules',

41
    ],

42
    keywords='machine learning',

43
)

44


45