dtunai / xLSTM-Jax

Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)
16Updated 5 months ago

Alternatives and similar repositories for xLSTM-Jax:

Users that are interested in xLSTM-Jax are comparing it to the libraries listed below