JPC: Flexible Inference for Predictive Coding Networks in JAX

Francesco Innocenti, Paul Kinghorn, Will Yun-Farmbrough +3

We introduce JPC, a JAX library for training neural networks with Predictive Coding. JPC provides a simple, fast and flexible interface to train a variety of PC networks (PCNs) inc…