setup.py

from setuptools import setup


setup(
        name='funlib.learn.jax',
        version='0.1',
        url='https://github.com/funkelab/funlib.learn.jax',
        author='Funkelab',
        author_email='funkej@janelia.hhmi.org',
        license='MIT',
        packages=[
            'funlib.learn.jax',
            'funlib.learn.jax.models'
        ]
)