Kozachkov, Leo

12 publications

TMLR 2026 A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems Xavier Gonzalez, E. Kelly Buchanan, Hyun Dong Lee, Jerry Weihong Liu, Ke Alexander Wang, David M. Zoltowski, Leo Kozachkov, Christopher Re, Scott Linderman

ICLR 2026 InputDSA: Demixing, Then Comparing Recurrent and Externally Driven Dynamics Ann Huang, Mitchell Ostrow, Satpreet Harcharan Singh, Leo Kozachkov, Ila R Fiete, Kanaka Rajan

ICLR 2026 NRGPT: An Energy-Based Alternative for GPT Nima Dehmamy, Benjamin Hoover, Bishwajit Saha, Leo Kozachkov, Jean-Jacques Slotine, Dmitry Krotov

NeurIPS 2025 Characterizing Control Between Interacting Subsystems with Deep Jacobian Estimation Adam Joseph Eisen, Mitchell Ostrow, Sarthak Chandra, Leo Kozachkov, Earl K Miller, Ila R Fiete

TMLR 2025 Generalized Orders of Magnitude for Scalable, Parallel, High-Dynamic-Range Computation Franz A. Heinsen, Leo Kozachkov

NeurIPS 2025 Intrinsic Goals for Autonomous Agents: Model-Based Exploration in Virtual Zebrafish Predicts Ethological Behavior and Whole-Brain Dynamics Reece Keller, Alyn Kirsch, Felix Pei, Xaq Pitkow, Leo Kozachkov, Aran Nayebi

NeurIPS 2025 Parallelizing MCMC Across the Sequence Length David M. Zoltowski, Skyler Wu, Xavier Gonzalez, Leo Kozachkov, Scott Linderman

NeurIPS 2025 Predictability Enables Parallelization of Nonlinear State Space Models Xavier Gonzalez, Leo Kozachkov, David M. Zoltowski, Kenneth L. Clarkson, Scott Linderman

NeurIPSW 2024 Neuron-Astrocyte Associative Memory Leo Kozachkov, Jean-Jacques Slotine, Dmitry Krotov

NeurIPS 2023 Beyond Geometry: Comparing the Temporal Structure of Computation in Neural Circuits with Dynamical Similarity Analysis Mitchell Ostrow, Adam Eisen, Leo Kozachkov, Ila Fiete

TMLR 2023 Generalization as Dynamical Robustness--the Role of Riemannian Contraction in Supervised Learning Leo Kozachkov, Patrick Wensing, Jean-Jacques Slotine

NeurIPS 2022 RNNs of RNNs: Recursive Construction of Stable Assemblies of Recurrent Neural Networks Leo Kozachkov, Michaela Ennis, Jean-Jacques Slotine