Unofficial implementation of Forward-Forward algorithm by jax.
Paper
-
Layer and Models
-
Losses and Optimizers
-
Preprocess codes for supervised model
-
Train codes of supervised model
-
Preprocess codes for unsupervised model
-
Train codes of unsupervised model
-
Online training codes
-
Fix layers to be parallelized.