GitHub repository: test-time-training/ttt-lm-jax
Primary language: Python
Description: Official JAX implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States