sensitivity_jax/setup.py at main · rdyro/sensitivity_jax · 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
import os
from setuptools import setup

# borrowed from https://pythonhosted.org/an_example_pypi_project/setuptools.html


def read(fname):
    try:
        return open(os.path.join(os.path.dirname(__file__), fname)).read()
    except FileNotFoundError:
        return ""


setup(
    name="sensitivity_jax",
    version="0.3.0",
    author="Robert Dyro",
    description=(
        "Optimization Sensitivity Analysis for Bilevel Programming for JAX"
    ),
    license="MIT",
    packages=["sensitivity_jax", "sensitivity_jax.extras"],
    install_requires=[
        "jax",
        "jaxlib",
        "torch",
        "tensorboard",
        "numpy",
        "scipy",
        "tqdm",
        "matplotlib",
        "joblib",
    ],
    long_description=read("README.md"),
)