A research-focused modular generative modeling library built on JAX/Flax NNX
AI agent memory system powered by Modern Hopfield Networks — no LLM calls, no database, one matrix multiply