├── .gitignore ├── LICENSE ├── README.md ├── main.py ├── parallax ├── __init__.py ├── core.py └── layers.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/main.py -------------------------------------------------------------------------------- /parallax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/parallax/__init__.py -------------------------------------------------------------------------------- /parallax/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/parallax/core.py -------------------------------------------------------------------------------- /parallax/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/parallax/layers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | jax 2 | jaxlib 3 | frozendict 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/srush/parallax/HEAD/setup.py --------------------------------------------------------------------------------