GitHub repository: 0xdti/xLSTM-Jax
Primary language: Python
Description: Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)