test-time-training / ttt-lm-jax

Official JAX implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

test-time-training/ttt-lm-jax Stargazers