├── .flake8 ├── .gitignore ├── LICENSE ├── README.md ├── allegro_jax ├── __init__.py ├── allegro.py ├── allegro_flax.py └── allegro_haiku.py ├── setup.cfg ├── setup.py └── test.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | *.egg-info 3 | *.ipynb 4 | .vscode -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/README.md -------------------------------------------------------------------------------- /allegro_jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/allegro_jax/__init__.py -------------------------------------------------------------------------------- /allegro_jax/allegro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/allegro_jax/allegro.py -------------------------------------------------------------------------------- /allegro_jax/allegro_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/allegro_jax/allegro_flax.py -------------------------------------------------------------------------------- /allegro_jax/allegro_haiku.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/allegro_jax/allegro_haiku.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mariogeiger/allegro-jax/HEAD/test.py --------------------------------------------------------------------------------