{
    "info": {
        "author": "Kakao Brain",
        "author_email": "",
        "bugtrack_url": null,
        "classifiers": [
            "Development Status :: 3 - Alpha",
            "Intended Audience :: Science/Research",
            "License :: OSI Approved :: Apache Software License",
            "Programming Language :: Python :: 3",
            "Programming Language :: Python :: 3.6",
            "Programming Language :: Python :: 3.7",
            "Programming Language :: Python :: 3 :: Only",
            "Topic :: Scientific/Engineering :: Artificial Intelligence",
            "Typing :: Typed"
        ],
        "description": "torchgpipe\n==========\n\nA GPipe_ implementation in PyTorch_.\n\n.. _GPipe: https://arxiv.org/abs/1811.06965\n.. _PyTorch: https://pytorch.org/\n\n.. sourcecode:: python\n\n   from torchgpipe import GPipe\n\n   model = nn.Sequential(a, b, c, d)\n   model = GPipe(model, balance=[1, 1, 1, 1], chunks=8)\n\n   for input in data_loader:\n       output = model(input)\n\nWhat is GPipe?\n~~~~~~~~~~~~~~\n\nGPipe is a scalable pipeline parallelism library published by Google Brain,\nwhich allows efficient training of large, memory-consuming models. According to\nthe paper, GPipe can train a 25x larger model by using 8x devices (TPU), and\ntrain a model 3.5x faster by using 4x devices.\n\n`GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism\n<https://arxiv.org/abs/1811.06965>`_\n\nGoogle trained AmoebaNet-B with 557M parameters over GPipe. This model has\nachieved 84.3% top-1 and 97.0% top-5 accuracy on ImageNet classification\nbenchmark (the state-of-the-art performance as of May 2019).\n\nLinks\n~~~~~\n\n- Source Code: https://github.com/kakaobrain/torchgpipe\n- Documentation: https://torchgpipe.readthedocs.io/\n- Original Paper: https://arxiv.org/abs/1811.06965",
        "description_content_type": "",
        "docs_url": null,
        "download_url": "",
        "downloads": {
            "last_day": -1,
            "last_month": -1,
            "last_week": -1
        },
        "home_page": "https://github.com/kakaobrain/torchgpipe",
        "keywords": "pytorch gpipe",
        "license": "Apache License 2.0",
        "maintainer": "Heungsub Lee, Myungryong Jeong, Chiheon Kim",
        "maintainer_email": "",
        "name": "torchgpipe",
        "package_url": "https://pypi.org/project/torchgpipe/",
        "platform": "",
        "project_url": "https://pypi.org/project/torchgpipe/",
        "project_urls": {
            "Homepage": "https://github.com/kakaobrain/torchgpipe"
        },
        "release_url": "https://pypi.org/project/torchgpipe/0.0.4/",
        "requires_dist": null,
        "requires_python": "",
        "summary": "GPipe for PyTorch",
        "version": "0.0.4"
    },
    "last_serial": 5944514,
    "releases": {
        "0.0.1": [
            {
                "comment_text": "",
                "digests": {
                    "md5": "3c7f40e540e6d16444417a99eeab4b67",
                    "sha256": "9a840108ccaea2a8e54305291121fa50713ef7bb061516511916d102876f43e4"
                },
                "downloads": -1,
                "filename": "torchgpipe-0.0.1.tar.gz",
                "has_sig": false,
                "md5_digest": "3c7f40e540e6d16444417a99eeab4b67",
                "packagetype": "sdist",
                "python_version": "source",
                "requires_python": null,
                "size": 8816,
                "upload_time": "2019-06-26T05:12:09",
                "url": "https://files.pythonhosted.org/packages/b8/aa/aa34c9cd3fe0fe533c076f88099efb5b1fa239336167a31ad86e35246dd7/torchgpipe-0.0.1.tar.gz"
            }
        ],
        "0.0.1a0": [
            {
                "comment_text": "",
                "digests": {
                    "md5": "1e5835a9538fe24f8cc9ec207f799f6d",
                    "sha256": "bbecf8e085dc26ef380d9efe15321a761bcc29901311b51e70abd21883a508eb"
                },
                "downloads": -1,
                "filename": "torchgpipe-0.0.1a0.tar.gz",
                "has_sig": false,
                "md5_digest": "1e5835a9538fe24f8cc9ec207f799f6d",
                "packagetype": "sdist",
                "python_version": "source",
                "requires_python": null,
                "size": 8843,
                "upload_time": "2019-05-14T13:07:52",
                "url": "https://files.pythonhosted.org/packages/1e/f8/2daa5df4b7b8adfd592afb762860283b9cad6cd81564f09b4148470f508d/torchgpipe-0.0.1a0.tar.gz"
            }
        ],
        "0.0.2": [
            {
                "comment_text": "",
                "digests": {
                    "md5": "9809691b51e6943653572f75c0c3f290",
                    "sha256": "e859e784e03008f8b74996239073cb23cc8196a958432139c84db60d9f1aab49"
                },
                "downloads": -1,
                "filename": "torchgpipe-0.0.2.tar.gz",
                "has_sig": false,
                "md5_digest": "9809691b51e6943653572f75c0c3f290",
                "packagetype": "sdist",
                "python_version": "source",
                "requires_python": null,
                "size": 16390,
                "upload_time": "2019-06-26T05:48:21",
                "url": "https://files.pythonhosted.org/packages/c8/9b/9c06ac8970348b86af9c1d83795b5659f8afef99e06c1045f2a92f9b3d75/torchgpipe-0.0.2.tar.gz"
            }
        ],
        "0.0.3": [
            {
                "comment_text": "",
                "digests": {
                    "md5": "831e46d52d8baf164411aea33389cd0d",
                    "sha256": "a9de300173800e45783f3f17959a24deb339ccdebd0f17245bf29fda8fa0c165"
                },
                "downloads": -1,
                "filename": "torchgpipe-0.0.3.tar.gz",
                "has_sig": false,
                "md5_digest": "831e46d52d8baf164411aea33389cd0d",
                "packagetype": "sdist",
                "python_version": "source",
                "requires_python": null,
                "size": 20460,
                "upload_time": "2019-09-30T07:42:35",
                "url": "https://files.pythonhosted.org/packages/ef/e1/21172ef06303514a0204dbdd9eb3220cd51c9d08c0b4c17d888ba30691d8/torchgpipe-0.0.3.tar.gz"
            }
        ],
        "0.0.4": [
            {
                "comment_text": "",
                "digests": {
                    "md5": "b23c45c30727574850c5c67fae235667",
                    "sha256": "f6c6471b908ec203d8f069d3e487fff8f19407bf1bc5900ec95fcf3e529e3d2a"
                },
                "downloads": -1,
                "filename": "torchgpipe-0.0.4.tar.gz",
                "has_sig": false,
                "md5_digest": "b23c45c30727574850c5c67fae235667",
                "packagetype": "sdist",
                "python_version": "source",
                "requires_python": null,
                "size": 21636,
                "upload_time": "2019-10-08T12:16:59",
                "url": "https://files.pythonhosted.org/packages/43/46/f998436520d65988ba852aff2724e97fb72f27ad6fb90d4fd3a7c83c18ab/torchgpipe-0.0.4.tar.gz"
            }
        ]
    },
    "urls": [
        {
            "comment_text": "",
            "digests": {
                "md5": "b23c45c30727574850c5c67fae235667",
                "sha256": "f6c6471b908ec203d8f069d3e487fff8f19407bf1bc5900ec95fcf3e529e3d2a"
            },
            "downloads": -1,
            "filename": "torchgpipe-0.0.4.tar.gz",
            "has_sig": false,
            "md5_digest": "b23c45c30727574850c5c67fae235667",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": null,
            "size": 21636,
            "upload_time": "2019-10-08T12:16:59",
            "url": "https://files.pythonhosted.org/packages/43/46/f998436520d65988ba852aff2724e97fb72f27ad6fb90d4fd3a7c83c18ab/torchgpipe-0.0.4.tar.gz"
        }
    ]
}