Installation
Install directly from GitHub with:
pip install git+https://github.com/mariogeiger/allegro-jax
Usage
from allegro_jax import AllegroLayer
Install directly from GitHub with:
pip install git+https://github.com/mariogeiger/allegro-jax
from allegro_jax import AllegroLayer