jax-sparse/setup.py at main · CrawlScript/jax-sparse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# coding=utf-8

from setuptools import setup, find_packages

setup(
    name="jax_sparse",
    python_requires='>3.5.0',
    version="0.0.2",
    author="Jun Hu",
    author_email="hujunxianligong@gmail.com",
    packages=find_packages(
        exclude=[
            'benchmarks',
            'data',
            'demo',
            'dist',
            'doc',
            'docs',
            'logs',
            'models',
            'test'
        ]
    ),
    install_requires=[
        "jax >= 0.3.13",
        # "networkx >= 2.1",
        # "scipy >= 1.1.0",
        # "scikit-learn >= 0.22",
        # "ogb_lite >= 0.0.3",
        # "tqdm"
    ],
    extras_require={

    },
    description="Efficient and Friendly Sparse Matrix Library for JAX.",
    license="GNU General Public License v3.0 (See LICENSE)",
    long_description="Efficient and Friendly Sparse Matrix Library for JAX.",
    # long_description=open("README.rst", "r", encoding="utf-8").read(),
    url="https://github.com/CrawlScript/jax-sparse"
)