diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..38b6b6f --- /dev/null +++ b/.gitignore @@ -0,0 +1,10 @@ +__pycache__ +/.idea +/venv/ +*.egg-info +/build/ +/dist/ +.env +# setup.py # actually, commit the stub setup.py file so that users can pip3 install -e . +.hypothesis/ +/poetry.lock diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..ab185d3 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,16 @@ +[tool.poetry] +name = "jaxtorch" +version = "0.1.0" +description = "PyTorch-style interface for JAX" +authors = ["nshepperd "] + +[tool.poetry.dependencies] +python = "^3.7" +einops = "^0.3.0" +cbor2 = "^5.4.1" + +[tool.poetry.dev-dependencies] + +[build-system] +requires = ["poetry-core>=1.0.0"] +build-backend = "poetry.core.masonry.api" diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..0d60a5e --- /dev/null +++ b/setup.py @@ -0,0 +1,30 @@ +# -*- coding: utf-8 -*- +from setuptools import setup + +packages = \ +['jaxtorch', 'jaxtorch.nn'] + +package_data = \ +{'': ['*']} + +install_requires = \ +['cbor2>=5.4.1,<6.0.0', 'einops>=0.3.0,<0.4.0'] + +setup_kwargs = { + 'name': 'jaxtorch', + 'version': '0.1.0', + 'description': 'PyTorch-style interface for JAX', + 'long_description': None, + 'author': 'nshepperd', + 'author_email': 'nshepperd@gmail.com', + 'maintainer': None, + 'maintainer_email': None, + 'url': None, + 'packages': packages, + 'package_data': package_data, + 'install_requires': install_requires, + 'python_requires': '>=3.7,<4.0', +} + + +setup(**setup_kwargs)