mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
17 lines
517 B
Python
17 lines
517 B
Python
from setuptools import setup
|
|
from glob import glob
|
|
|
|
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',
|
|
package_data={'jax.lib': glob('jax/lib/*.so')},
|
|
)
|