2018-11-18 15:17:32 -08:00
|
|
|
from setuptools import setup
|
2018-11-18 15:28:22 -08:00
|
|
|
from glob import glob
|
2018-11-18 15:17:32 -08:00
|
|
|
|
|
|
|
|
setup(
|
|
|
|
|
name='jax',
|
|
|
|
|
version='0.0',
|
|
|
|
|
description='Differentiate, compile, and transform Numpy code.',
|
|
|
|
|
author='JAX team',
|
|
|
|
|
author_email='jax-team@google.com',
|
|
|
|
|
packages=['jax', 'jax.lib', 'jax.interpreters', 'jax.numpy', 'jax.scipy',
|
|
|
|
|
'jax.experimental'],
|
|
|
|
|
install_requires=['numpy>=1.12', 'six', 'protobuf'],
|
|
|
|
|
url='https://github.com/google/jax',
|
|
|
|
|
license='Apache-2.0',
|
2018-11-18 15:28:22 -08:00
|
|
|
package_data={'jax.lib': glob('jax/lib/*.so')},
|
2018-11-18 15:17:32 -08:00
|
|
|
)
|
