Ryu1845 / hyena-jax Public

Notifications You must be signed in to change notification settings
Fork 2
Star 10

Implementation of Hyena Hierarchy in JAX

Apache-2.0 license

10 stars 2 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
hyena_flax.py		hyena_flax.py
hyena_pytorch.py		hyena_pytorch.py

Repository files navigation

hyena-jax

Port of Hyena Hierarchy to JAX

Original: https://github.com/HazyResearch/safari/blob/main/standalone_hyena.py

Flax implementation : https://github.com/Ryu1845/hyena-jax/blob/main/hyena_flax.py

About

Implementation of Hyena Hierarchy in JAX

python nlp machine-learning flax hyena jax llm

Apache-2.0 license

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%