GradMem: Teaching LLMs to Remember (Without Retraining)

Send us Fan MailDiffusion models have become the foundation of modern generative AI, powering state-of-the-art systems for image generation, video synthesis, protein design, and more. In the first episode of this series, we introduced generative modeling as a sampling problem and developed the probabilistic framework needed to think about how models learn complex data distributions.In this second episode, we move into the core mathematics behind flow matching and diffusion modeling, continuing through the MIT lecture notes An Introduction to Flow Matching and Diffusion Models by Peter Holderrieth and Ezra Erives.We explore how a simple, easy-to-sample distribution such as Gaussian noise can be continuously transformed into a complicated data distribution, and how this idea leads naturally to both deterministic and stochastic approaches to generative modeling.Topics covered include:The basic idea behind flow matchingTransforming a noise distribution into a data distributionProbability paths and time-dependent distributionsVector fields and continuous-time generative processesOrdinary differential equations (ODEs) in generative modelingStochastic differential equations (SDEs) and the role of injected noiseThe relationship between diffusion models and continuous flowsForward and reverse diffusion processesHow trained diffusion and flow models are used during inferenceNumerical inference algorithms for solving the resulting ODEs and SDEsWhy the number of inference steps affects speed and generation qualityHow these mathematical formulations connect to practical modern diffusion systemsThe goal of this episode is to build intuition for what a diffusion model is actually doing mathematically: rather than simply “removing noise,” we can view generation as learning a dynamical process that transports probability from a simple starting distribution toward the complex distribution of real data.This framework provides the foundation for later discussions of score matching, denoising objectives, probability flow ODEs, modern sampling algorithms, and the architectures used in state-of-the-art generative models.Lecture Notes:https://diffusion.csail.mit.edu/docs/lecture-notes.pdfWebsite:https://arkitekt-ai.comContact:[email protected]

Send us Fan MailDiffusion models have become the foundation of modern generative AI, powering state-of-the-art systems for image generation, video synthesis, protein design, and more. But behind the impressive demos lies a surprisingly elegant mathematical framework.In this first episode of a multi-part series, we begin working through the excellent MIT lecture notes An Introduction to Flow Matching and Diffusion Models by Peter Holderrieth and Ezra Erives. Rather than jumping straight into denoising or neural networks, we focus on the fundamental question: What does it actually mean to generate data? Topics covered include: Why generative modeling is fundamentally a sampling problem Data distributions and probability densities  Representing images, videos, and other data as vectors  Conditional generation and prompts  Why diffusion models are framed as learning distributions rather than memorizing examples  An overview of where the mathematics is heading in the remainder of the course This episode is intended for anyone who wants to understand diffusion models from first principles, building the mathematical intuition needed for later discussions on ODEs, SDEs, flow matching, score matching, and modern diffusion architectures. Lecture Notes: https://diffusion.csail.mit.edu/docs/lecture-notes.pdfWebsite: https://arkitekt-ai.comContact: [email protected]

Send us Fan MailIn this episode, we explore the paper "Generative Recursive Reasoning (GRAM)," a fascinating new approach to AI reasoning co-authored by Yoshua Bengio and researchers from Mila and Samsung AI.Most modern AI systems reason by generating more tokens. GRAM takes a different approach: instead of extending a chain of thought, it repeatedly refines an internal latent state. The key innovation is introducing probabilistic reasoning trajectories, allowing the model to explore multiple possible solutions simultaneously rather than committing to a single deterministic path.We discuss:Recursive Reasoning Models (RRMs) and why they differ from traditional transformersThe limitations of deterministic latent reasoningHow GRAM introduces stochastic latent trajectoriesVariational inference and the roles of pθ and qϕMulti-hypothesis reasoning and inference-time scalingResults on Sudoku, ARC-AGI, N-Queens, and other structured reasoning benchmarksWhy latent-space reasoning may become an alternative to longer chain-of-thought promptingThe paper also demonstrates unconditional generation capabilities, suggesting a path toward reasoning systems that can both solve problems and generate structured outputs through recursive latent computation.PDF:Generative Recursive ReasoningArkitekt AI:https://arkitekt-ai.comContact:[email protected]

Send us Fan MailIn this episode, we break down the new paper “OccuBench: Evaluating AI Agents on Real-World Professional Tasks via Language Environment Simulation,” which explores how AI agents can be benchmarked across real occupational domains like healthcare, logistics, manufacturing, customs processing, and more.The paper introduces OccuBench, a large-scale benchmark spanning 100 professional task scenarios across 65 specialized domains. One of the most interesting ideas is the use of Language Environment Simulators (LESs), where LLMs simulate enterprise environments and tool responses for domains that normally have no public APIs or accessible evaluation environments.We discuss:Why current agent benchmarks miss most real-world enterprise workHow simulated environments can evaluate professional AI agentsFault injection testing and robustness evaluationCross-industry capability differences between frontier modelsWhat this means for autonomous enterprise systems and AI agents in productionPaper:https://arxiv.org/abs/2604.10866PDF:https://arxiv.org/pdf/2604.10866Arkitekt AI:arkitekt-ai.comContact:[email protected]

Send us Fan MailIn this episode, we break down GradMem, a new approach to memory in large language models: https://arxiv.org/pdf/2603.13875v1Instead of relying on the transformer KV cache or repeatedly reprocessing documents (like in RAG), GradMem introduces a different idea—learn a compact memory representation at inference time. Using a few steps of gradient descent, the model “writes” important information from a context into a small set of memory tokens, allowing it to answer future queries without needing the original context.We cover: Why KV cache is a brute-force solution to long context  How test-time optimization turns memory into something learnable  The difference between storing text vs. storing information  What this means for agents, RAG systems, and long-horizon tasks Big takeaway:Instead of reading context over and over, models can learn to compress and reuse it intelligently.Learn more / build with AIhttps://www.arkitekt-ai.com/

Send us Fan MailIn this episode, we break down a fascinating new result from recent research: that modern Transformer language models are almost surely injective—meaning different prompts map to unique internal representations, with no information loss.We dig into the paper:Read the paper on arXivAt the core of the proof is a surprisingly deep mathematical idea: Transformers are real analytic functions of their parameters, which allows researchers to rigorously reason about when “collisions” (two prompts producing the same representation) can occur. The result? Collisions only happen on a measure zero set—mathematically possible, but practically never observed. We unpack:What it means for a function to be real analyticWhy this implies near-perfect uniqueness of representationsHow gradient descent preserves this property during trainingAnd what this says about interpretability, privacy, and reversibility of LLMsWe also explore the practical implications—if models are truly invertible, could we reconstruct inputs from activations? What does that mean for safety and data leakage?About the HostThis episode is brought to you by Arkitekt AI — an automated enterprise software development platform that builds full analytics, ML, and data systems from natural language.Learn more: https://arkitekt-ai.com

Send us Fan MailLink to arxiv: https://arxiv.org/pdf/2602.04118Large language models have recently shown impressive reasoning abilities, often learned through reinforcement learning and low-rank adaptation techniques like LoRA. But these approaches still assume that effective reasoning requires relatively large adaptation layers. This new paper challenges that assumption by asking a provocative question: how small can a reasoning update really be?In this episode, we explore Learning to Reason in 13 Parameters, which introduces TinyLoRA, a method that compresses low-rank adapters down to the extreme — in some cases to just a single parameter. Instead of relying on large adaptation matrices, TinyLoRA demonstrates that reasoning behavior can be steered using ultra-minimal parameter updates, dramatically reducing the computational and memory footprint required to teach models new reasoning skills. We break down:Why conventional LoRA and low-rank adapters hit a floor at model dimensionality,How TinyLoRA scales reasoning adapters down to near-zero parameter counts,What this reveals about where reasoning ability actually lives inside neural networks,And why tiny adaptation layers could reshape efficient fine-tuning, on-device intelligence, and rapid deployment.The results suggest that reasoning competence may not require massive structural changes — only precisely targeted parameter nudges. This challenges assumptions about scaling, efficiency, and the true complexity of learned reasoning.

Send us Fan MailLarge Language Models often struggle with complex planning tasks that require exploration, backtracking, and self-correction. Once an LLM commits to an early mistake, its linear chain-of-thought reasoning makes recovery difficult. While search methods like Monte Carlo Tree Search (MCTS) offer a way to explore alternatives, they typically rely on sparse rewards and fail to fully exploit the semantic strengths of language models.In this episode, we dive into SPIRAL (Symbolic LLM Planning via Grounded and Reflective Search), a new framework that fundamentally rethinks how planning and search interact in LLM-based agents. Instead of treating MCTS as a brute-force optimizer, SPIRAL embeds a cognitive architecture of three specialized LLM roles directly into the search loop:A Planner proposes creative next actions,A Simulator grounds those actions by predicting realistic outcomes, andA Critic reflects on the results to provide dense, informative reward signals.This planner–simulator–critic loop transforms search into a guided, self-correcting reasoning process, allowing agents to recover from mistakes, evaluate alternatives more effectively, and plan with far greater robustness.Paper link: https://arxiv.org/pdf/2512.23167Repo: https://github.com/IBM/SPIRAL

Send us Fan MailEpisode Paper: https://arxiv.org/pdf/2512.16848In this episode, we dive into a cutting-edge AI research breakthrough that tackles one of the biggest challenges in training intelligent agents: how to explore effectively. Standard reinforcement learning (RL) methods help language model agents learn to interact with environments and solve multi-step tasks, but they often struggle when the tasks require active exploration—that is, learning what to try next when the best strategy isn’t obvious from past experience. The new paper introduces LaMer, a Meta-Reinforcement Learning (Meta-RL) framework designed to give language agents the ability to learn how to explore. Unlike conventional RL agents that learn a fixed policy, LaMer’s Meta-RL approach encourages agents to flexibly adapt by learning from their own trial-and-error experiences. This means agents can better adapt to novel or more difficult environments without needing massive retraining. We’ll explain:Why exploration is critical for long-horizon tasks with delayed or sparse rewards. How Meta-RL shifts the focus from fixed policies to adaptable exploration behavior. What LaMer’s results suggest about learned exploration and generalization in AI systems. Whether you’re into reinforcement learning, multi-agent systems, or the future of adaptive AI, this episode breaks down how Meta-RL could help agents think more like explorers—not just pattern followers.

Send us Fan MailIn this episode, we unpack DeepSearch, a new paradigm in reinforcement learning with verifiable rewards (RLVR) that aims to overcome one of the biggest bottlenecks in training reasoning-capable AI systems. Traditional reinforcement learning methods often plateau after extensive training because they rely on sparse exploration and limited rollouts, leaving critical reasoning paths undiscovered and unlearned.DeepSearch turns this model training approach on its head by embedding Monte Carlo Tree Search (MCTS) directly into the training loop—not just at inference time. This fundamentally changes how models explore the space of possible solutions: instead of brute-force parameter scaling or longer training runs, DeepSearch uses structured, systematic exploration to dramatically improve learning efficiency.We break down how DeepSearch:Injects tree search into training, enabling richer exploration of reasoning paths.Uses a global frontier strategy to prioritize promising reasoning trajectories.Improves training-time credit assignment, so models learn not only from success but from strategic exploration itself.Achieves impressive results on benchmarks for mathematical reasoning, setting new state-of-the-art performance and using fewer computational resources. Whether you’re a machine learning researcher, an AI enthusiast, or just curious about the future of intelligent systems, this episode explores how search-augmented learning could redefine how future AI systems master complex reasoning problems.DeepSearch: Overcome the Bottleneck of Reinforcement Learning with Verifiable Rewards via Monte Carlo Tree Search