A transformer language model rebuilt around predictive coding. The embeddings, attention,
MLP, and output layers each maintain a latent state, predict the activity of the next
layer, and update from their own prediction error over T inference steps.
I led the research and built the training stack, including the model architecture,
multi-GPU training, KV caching, a two-phase hyperparameter search, and experiment
visualizations.
Case study·Code
An independent replication of the NGC framework from Ororbia and Kifer's
Nature Communications paper. I rebuilt the evaluation around reconstruction,
likelihood and downstream classification, added a masked-MSE pattern completion probe,
benchmarked against backpropagation-trained baselines, and packaged the whole comparison
so it can be re-run from a clean machine.
Case study·Code·Paper
A JAX library from SingularityNET for building predictive coding networks, designed
to train the same architecture with either predictive coding or backpropagation for
direct comparison. I contributed transformer components, regularization techniques,
and BPE tokenization, along with a demo for training a simple model using both
predictive coding and backpropagation.
Case study·Code