JPC: Flexible Inference for Predictive Coding Networks in JAX

The free energy principle for action and perception: A mathematical review

C. L. Buckley, C. S. Kim, S. McGregor, and A. K. Seth · 2017

Earlier work this paper cites.

The DeepMind JAX Ecosystem, 2020

DeepMind, I. Babuschkin, K. Baumli, A. Bell, S. Bhupatiraju, J. Bruce, P. Buchlovsky, D. Budden, T. Cai, A. Clark, I. Danihelka, A. Dedieu, C. Fantacci, J. Godwin, C. Jones, R. Hemsley, T. Hennigan, M. Hessel, S. Hou, S. Kapturowski, T. Keck, I. Kemaev, M. King, M. Kunesch, L. Martens, H. Merzic, V. Mikulik, T. Norman, G. Papamakarios, J. Quan, R. Ring, F. Ruiz, A. Sanchez, L. Sartran, R. Schneider, E. Sezener, S. Spencer, S. Srinivasan, M. Stanojević, W. Stokowiec, L. Wang, G. Zhou, and F. Viola · 2020

Earlier work this paper cites.

Equinox: neural networks in jax via callable pytrees and filtered transformations

Original

P. Kidger and C. Garcia · 2021

Earlier work this paper cites.

Predictive coding: a theoretical and experimental review

Original

B. Millidge, A. Seth, and C. L. Buckley · 2021

Earlier work this paper cites.

On neural differential equations

Original

P. Kidger · 2022

Earlier work this paper cites.

Predictive coding: towards a future of deep learning beyond backpropagation?

Original

B. Millidge, T. Salvatori, Y. Song, R. Bogacz, and T. Lukasiewicz · 2022

Cited alongside, same era.

On the relationship between predictive coding and backpropagation

R. Rosenbaum · 2022

Cited alongside, same era.

Brain-inspired computational intelligence via predictive coding

Original

T. Salvatori, A. Mali, C. L. Buckley, T. Lukasiewicz, R. P. Rao, K. Friston, and A. Ororbia · 2023

Cited alongside, same era.

Hybrid predictive coding: Inferring, fast and slow

A. Tscshantz, B. Millidge, A. K. Seth, and C. L. Buckley · 2023

Cited alongside, same era.

Sample as you infer: Predictive coding with langevin dynamics

Original

U. Zahid, Q. Guo, and Z. Fountas · 2023

Cited alongside, same era.

Only strict saddles in the energy landscape of predictive coding networks?

Original

F. Innocenti, E. M. Achour, R. Singh, and C. L. Buckley · 2024

Closest in time.

pyhgf: A neural network library for predictive coding

Original

N. Legrand, L. Weber, P. T. Waade, A. H. M. Daugaard, M. Khodadadi, N. Mikuš, and C. Mathys · 2024

Closest in time.

Learning probability distributions of sensory inputs with monte carlo predictive coding

G. Oliviers, R. Bogacz, and A. Meulemans · 2024

Closest in time.

Benchmarking predictive coding networks–made simple

Original

L. Pinchetti, C. Qi, O. Lokshyn, G. Olivers, C. Emde, M. Tang, A. M’Charrak, S. Frieder, B. Menzat, R. Bogacz, et al · 2024

Closest in time.

Predictive coding networks and inference learning: Tutorial and survey

Original