https://github.com/mariogeiger/allegro-jax

https://github.com/mariogeiger/allegro-jax

Last synced: 15 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/mariogeiger/allegro-jax
• Owner: mariogeiger
• Created: 2023-07-02T19:00:00.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-04-09T18:44:30.000Z (7 months ago)
• Last Synced: 2024-10-12T05:52:24.881Z (about 1 month ago)
• Language: Python
• Size: 9.77 KB
• Stars: 17
• Watchers: 2
• Forks: 2
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• best-of-atomistic-machine-learning - GitHub - 50% open · ⏱️ 09.04.2024): (Interatomic Potentials (ML-IAP))

README

        
## Installation

Install directly from GitHub with:

```

pip install git+https://github.com/mariogeiger/allegro-jax

```

## Usage

You can import the Allegro layer directly with:

```python

from allegro_jax import AllegroLayer # For Flax

from allegro_jax import AllegroHaikuLayer # For Haiku

```

You can also import the whole model with:

```python

from allegro_jax import Allegro # For Flax

from allegro_jax import AllegroHaiku # For Haiku

```

See `test.py` for an example of usage.