dtunai / xLSTM-Jax

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

Related projects

Alternatives and complementary repositories for xLSTM-Jax