S4 Replication
Summary
A port of S4 to JAX + Flax NNX for learning purposes. Largely followed the Annotated S4 guide. No experiment reproductions have been attempted yet.
Select a result to preview
A port of S4 to JAX + Flax NNX for learning purposes. Largely followed the Annotated S4 guide. No experiment reproductions have been attempted yet.