FermiQ / JAX-Forward-Forward

JAX: Autograd and XLA: The Forward-Forward

FF-jax

Unofficial implementation of Forward-Forward algorithm by jax.
Paper

TODO

• 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.