test-time-training / ttt-lm-jax

Official JAX implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States
384Updated 5 months ago

Alternatives and similar repositories for ttt-lm-jax:

Users that are interested in ttt-lm-jax are comparing it to the libraries listed below