setup.py

from setuptools import setup

setup(
    name='wavetorch',
    version='0.2.1',
    description='Numerically solving and backpropagating through the wave equation using pytorch',
    url='https://github.com/fancompute/wavetorch',
    author='Ian Williamson, Tyler Hughes, Momchil Minkov, Shanhui Fan',
    author_email='ian.williamson@ieee.org',
    license='MIT',
    packages=['wavetorch'],
    scripts=[],
    zip_safe=False)