← Search

Leo Kozachkov

8 accepted papers

2026

InputDSA: Demixing, then comparing recurrent and externally driven dynamics

ICLR 2026poster

In control problems and basic scientific modeling, it is important to compare observations with dynamical simulations. For example, comparing two neural systems can shed light on the nature of emergent computations in the brain and deep neural networks. Recently, Ostrow et al. (2023) introduced Dyn…

Cited by 0SourceScholar
2026

NRGPT: An Energy-based Alternative for GPT

ICLR 2026poster

Generative Pre-trained Transformer (GPT) architectures are the most popular design for language modeling. Energy-based modeling is a different paradigm that views inference as a dynamical process operating on an energy landscape. We propose a minimal modification of the GPT setting to unify it with…

Cited by 0SourceScholar
2025

Characterizing control between interacting subsystems with deep Jacobian estimation

NeurIPS 2025spotlight

Biological function arises through the dynamical interactions of multiple subsystems, including those between brain areas, within gene regulatory networks, and more. A common approach to understanding these systems is to model the dynamics of each subsystem and characterize communication between the…

Cited by 0SourceScholar
2025

Intrinsic Goals for Autonomous Agents: Model-Based Exploration in Virtual Zebrafish Predicts Ethological Behavior and Whole-Brain Dynamics

NeurIPS 2025poster

Autonomy is a hallmark of animal intelligence, enabling adaptive and intelligent behavior in complex environments without relying on external reward or task structure. Existing reinforcement learning approaches to exploration in reward-free environments, including a class of methods known as *model-…

Cited by 0SourceScholar
2025

Parallelizing MCMC Across the Sequence Length

NeurIPS 2025poster

Markov chain Monte Carlo (MCMC) methods are foundational algorithms for Bayesian inference and probabilistic modeling. However, most MCMC algorithms are inherently sequential and their time complexity scales linearly with the sequence length. Previous work on adapting MCMC to modern hardware has the…

Cited by 0SourceScholar
2025

Predictability Enables Parallelization of Nonlinear State Space Models

NeurIPS 2025poster

The rise of parallel computing hardware has made it increasingly important to understand which nonlinear state space models can be efficiently parallelized. Recent advances have shown that evaluating a state space model can be recast as solving a parallelizable optimization problem, and sometimes th…

Cited by 0SourceScholar
2023

Beyond Geometry: Comparing the Temporal Structure of Computation in Neural Circuits with Dynamical Similarity Analysis

NeurIPS 2023poster

How can we tell whether two neural networks utilize the same internal processes for a particular computation? This question is pertinent for multiple subfields of neuroscience and machine learning, including neuroAI, mechanistic interpretability, and brain-machine interfaces. Standard approaches for…

2022

RNNs of RNNs: Recursive Construction of Stable Assemblies of Recurrent Neural Networks

NeurIPS 2022accept

Recurrent neural networks (RNNs) are widely used throughout neuroscience as models of local neural activity. Many properties of single RNNs are well characterized theoretically, but experimental neuroscience has moved in the direction of studying multiple interacting areas, and RNN theory needs to b…