evanatyourservice / xLSTM-JAX Public

Notifications You must be signed in to change notification settings
Fork 0
Star 3

An implementation of mLSTM from xLSTM in JAX

3 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
mlstm_cell.py		mlstm_cell.py
mlstm_layer.py		mlstm_layer.py
slstm.py		slstm.py

Repository files navigation

xLSTM-JAX

Simple implementations of mLSTM and sLSTM from xLSTM in JAX

Paper: https://arxiv.org/abs/2405.04517
Original implementation: https://github.com/NX-AI/xlstm

About

An implementation of mLSTM from xLSTM in JAX

lstm rnn flax jax xlstm

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%