NACLab · ago109 · Apr 12, 2025 · Mar 17, 2025 · Mar 21, 2025 · Mar 22, 2025
diff --git a/README.md b/README.md
@@ -2,19 +2,15 @@
 
 <img src="docs/images/ngc-learn-logo.png" width="300">
 
-<b>ngc-learn</b> is a Python library for building, simulating, and analyzing
-biomimetic systems, neurobiological agents, spiking neuronal networks,
-predictive coding circuitry, and models that learn via biologically-plausible
-forms of credit assignment. This simulation toolkit is built on top of JAX and is
-distributed under the 3-Clause BSD license.
+<b>ngc-learn</b> is a Python library for building, simulating, and analyzing biophysical / neurobiological systems, spiking neuronal networks, predictive coding circuitry, and biomimetic (NeuroAI) agents that learn in a biologically-plausible manner. This simulation toolkit, meant to support computational neuroscience and brain-inspired computing research, is built on top of JAX and is distributed under the 3-Clause BSD license.
 
 It is currently maintained by the
 <a href="https://www.cs.rit.edu/~ago/nac_lab.html">Neural Adaptive Computing (NAC) laboratory</a>.
 
 ## <b>Documentation</b>
 
 Official documentation, including tutorials, can be found
-<a href="https://ngc-learn.readthedocs.io/en/latest/#">here</a>. The model museum repo,
+<a href="https://ngc-learn.readthedocs.io/en/latest/#">here</a>. The model museum repo (ngc-museum),
 which implements several historical models, can be found
 <a href="https://github.com/NACLab/ngc-museum">here</a>.
 
@@ -36,8 +32,8 @@ ngc-learn requires:
 1) Python (>=3.10)
 2) NumPy (>=1.26.0)
 3) SciPy (>=1.7.0)
-4) ngcsimlib (>=0.3.b4), (visit official page <a href="https://github.com/NACLab/ngc-sim-lib">here</a>)
-5) JAX (>= 0.4.28) (to enable GPU use, make sure to install one of the CUDA variants)
+4) ngcsimlib (>=1.0.0), (visit official page <a href="https://github.com/NACLab/ngc-sim-lib">here</a>)
+5) JAX (>=0.4.28) (to enable GPU use, make sure to install one of the CUDA variants)
 <!--
 5) scikit-learn (>=1.3.1) if using `ngclearn.utils.density`
 6) matplotlib (>=3.4.3) if using `ngclearn.utils.viz`
@@ -46,7 +42,7 @@ ngc-learn requires:
 -->
 
 ---
-ngc-learn 1.2.beta2 and later require Python 3.10 or newer as well as ngcsimlib >=0.3.b4.
+ngc-learn 2.0.0 and later require Python 3.10 or newer as well as ngcsimlib >=1.0.0.
 ngc-learn's plotting capabilities (routines within `ngclearn.utils.viz`) require
 Matplotlib (>=3.8.0) and imageio (>=2.31.5) and both plotting and density estimation
 tools (routines within ``ngclearn.utils.density``) will require Scikit-learn (>=0.24.2).
@@ -66,7 +62,7 @@ running the above pip command if you want to use the GPU version.
 
 The documentation includes more detailed
 <a href="https://ngc-learn.readthedocs.io/en/latest/installation.html">installation instructions</a>.
-Note that this library was developed on Ubuntu 20.04 and tested on Ubuntu(s) 18.04 and 20.04.
+Note that this library was developed on Ubuntu 20.04/22.04 and tested on Ubuntu(s) 20.04 and 22.04.
 
 If the installation was successful, you should see the following if you test
 it against your Python interpreter, i.e., run the <code>$ python</code> command

diff --git a/docs/museum/snn_bfa.md b/docs/museum/snn_bfa.md
@@ -233,7 +233,7 @@ of $1000$ `SLIF` cells) similar to the one below:
 <img src="../images/museum/bfa_snn/bfasnn_codes.jpg" width="450" />
 
 Intriguingly, we see that the latent codes represented by the BFA-SNN's hidden
-layer spikes yield a rather (piecewise) linearly-separable transformation
+layer spikes yield a rather (piecewise) linearly-separable representation
 of the input digits, making the process of mapping inputs to label vectors
 much easier for the model's second layer of classification LIF units.
 Note that, in the `BFA_SNN` model exhibit class, we estimated

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -5,7 +5,7 @@ numpy>=1.26.0
 scikit-learn>=0.24.2
 scipy>=1.7.0
 matplotlib>=3.8.0
-jax>=0.4.18
-jaxlib>=0.4.18
+jax>=0.4.28
+jaxlib>=0.4.28
 imageio>=2.31.5
-ngcsimlib>=0.3.b4
+ngcsimlib>=1.0.0
diff --git a/docs/tutorials/model_basics/evolving_synapses.md b/docs/tutorials/model_basics/evolving_synapses.md
@@ -19,7 +19,7 @@ We do this specifically as follows:
 ```python
 from jax import numpy as jnp, random, jit
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 from ngclearn.components import HebbianSynapse, RateCell
 import ngclearn.utils.weight_distribution as dist
 
@@ -49,16 +49,16 @@ with Context("Circuit") as circuit:
   Wab.post << b.zF
 
   ## create and compile core simulation commands  
-  evolve_process = (Process()
+  evolve_process = (JaxProcess()
                     >> a.evolve)
   circuit.wrap_and_add_command(jit(evolve_process.pure), name="evolve")
 
-  advance_process = (Process()
+  advance_process = (JaxProcess()
                      >> a.advance_state)
   circuit.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-  reset_process = (Process()
-                 >> a.reset)
+  reset_process = (JaxProcess()
+                   >> a.reset)
   circuit.wrap_and_add_command(jit(reset_process.pure), name="reset")
 
   ## set up non-compiled utility commands
@@ -83,7 +83,6 @@ for ts in range(x_seq.shape[1]):
   circuit.advance(t=ts*1., dt=1.)
   circuit.evolve(t=ts*1., dt=1.)
   print(" {}: input = {} ~> Wab = {}".format(ts, x_t, Wab.weights.value))
-
 ```
 
 Your code should produce the same output (towards the bottom):

diff --git a/docs/tutorials/model_basics/model_building.md b/docs/tutorials/model_basics/model_building.md
@@ -10,6 +10,7 @@ While building our dynamical system we will set up a Context and then add the th
 ```python
 from jax import numpy as jnp, random
 from ngclearn import Context
+from ngclearn.utils import JaxProcess
 from ngcsimlib.compilers.process import Process
 from ngclearn.components import RateCell, HebbianSynapse
 import ngclearn.utils.weight_distribution as dist
@@ -71,13 +72,13 @@ This is simply done with the use of the following convenience function calls:
 
 ```python
     ## configure desired commands for simulation object
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> a.reset
                      >> Wab.reset
                      >> b.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> a.advance_state
                        >> Wab.advance_state
                        >> b.advance_state)

diff --git a/docs/tutorials/neurocog/adex_cell.md b/docs/tutorials/neurocog/adex_cell.md
@@ -24,7 +24,7 @@ import numpy as np
 
 from ngclearn.utils.model_utils import scanner
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.spiking.adExCell import AdExCell
 
@@ -48,11 +48,11 @@ with Context("Model") as model:
     )
 
     ## create and compile core simulation commands
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/error_cell.md b/docs/tutorials/neurocog/error_cell.md
@@ -54,7 +54,7 @@ The code you would write amounts to the below:
 ```python
 from jax import numpy as jnp, jit
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process, transition
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.graded.gaussianErrorCell import GaussianErrorCell
 
@@ -64,11 +64,11 @@ T = 5  ## number time steps to simulate
 with Context("Model") as model:
     cell = GaussianErrorCell("z0", n_units=3)
 
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/fitzhugh_nagumo_cell.md b/docs/tutorials/neurocog/fitzhugh_nagumo_cell.md
@@ -18,7 +18,7 @@ from jax import numpy as jnp, random, jit
 import numpy as np
 
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.spiking.fitzhughNagumoCell import FitzhughNagumoCell
 
@@ -40,11 +40,11 @@ with Context("Model") as model:
                               gamma=gamma, v0=v0, w0=w0, integration_type="euler")
 
     ## create and compile core simulation commands
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/hebbian.md b/docs/tutorials/neurocog/hebbian.md
@@ -38,11 +38,11 @@ Wab.post << b.zF
 as well as (a bit later in the model construction code):
 
 ```python
-evolve_process = (Process()
+evolve_process = (JaxProcess()
                   >> a.evolve)
 circuit.wrap_and_add_command(jit(evolve_process.pure), name="evolve")
 
-advance_process = (Process()
+advance_process = (JaxProcess()
                    >> a.advance_state)
 circuit.wrap_and_add_command(jit(advance_process.pure), name="advance")
 ```

diff --git a/docs/tutorials/neurocog/hodgkin_huxley_cell.md b/docs/tutorials/neurocog/hodgkin_huxley_cell.md
@@ -24,7 +24,7 @@ import numpy as np
 
 from ngclearn.utils.model_utils import scanner
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.spiking.hodgkinHuxleyCell import HodgkinHuxleyCell
 
@@ -52,11 +52,11 @@ with Context("Model") as model:
     )
 
     ## create and compile core simulation commands
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/input_cells.md b/docs/tutorials/neurocog/input_cells.md
@@ -40,7 +40,7 @@ spike train over $100$ steps in time as follows:
 ```python
 from jax import numpy as jnp, random, jit
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 
 from ngclearn.utils.viz.raster import create_raster_plot
 ## import model-specific mechanisms
@@ -56,11 +56,11 @@ T = 100  ## number time steps to simulate
 with Context("Model") as model:
     cell = BernoulliCell("z0", n_units=10, key=subkeys[0])
 
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/izhikevich_cell.md b/docs/tutorials/neurocog/izhikevich_cell.md
@@ -20,7 +20,7 @@ from jax import numpy as jnp, random, jit
 import numpy as np
 
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.spiking.izhikevichCell import IzhikevichCell
 
@@ -44,11 +44,11 @@ with Context("Model") as model:
                           integration_type="euler", v0=v0, w0=w0, key=subkeys[0])
 
     ## create and compile core simulation commands
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/lif.md b/docs/tutorials/neurocog/lif.md
@@ -25,7 +25,7 @@ cell, you would write code akin to the following:
 from jax import numpy as jnp, random, jit
 
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components.neurons.spiking.LIFCell import LIFCell
 from ngclearn.utils.viz.spike_plot import plot_spiking_neuron
@@ -47,11 +47,11 @@ with Context("Model") as model:
                    refract_time=2., key=subkeys[0])
 
     ## create and compile core simulation commands
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/mod_stdp.md b/docs/tutorials/neurocog/mod_stdp.md
@@ -42,7 +42,7 @@ and the required compiled simulation and dynamic commands, can be done as follow
 ```python 
 from jax import numpy as jnp, random, jit
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components import (TraceSTDPSynapse, MSTDPETSynapse,
                                  RewardErrorCell, VarTrace)
@@ -75,13 +75,13 @@ with Context("Model") as model:
     tr1 = VarTrace("tr1", n_units=1, tau_tr=tau_post, a_delta=Aminus)
     rpe = RewardErrorCell("r", n_units=1, alpha=0.)
 
-    evolve_process = (Process()
+    evolve_process = (JaxProcess()
                       >> W_stdp.evolve
                       >> W_mstdp.evolve
                       >> W_mstdpet.evolve)
     model.wrap_and_add_command(jit(evolve_process.pure), name="evolve")
 
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> tr0.advance_state
                        >> tr1.advance_state
                        >> rpe.advance_state
@@ -90,7 +90,7 @@ with Context("Model") as model:
                        >> W_mstdpet.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> W_stdp.reset
                      >> W_mstdp.reset
                      >> W_mstdpet.reset

diff --git a/docs/tutorials/neurocog/rate_cell.md b/docs/tutorials/neurocog/rate_cell.md
@@ -16,7 +16,7 @@ specifically the rate-cell (RateCell). Let's start with the file's header
 
 ```python
 from jax import numpy as jnp, random, jit
-from ngcsimlib.compilers.process import Process, transition
+from ngclearn.utils import JaxProcess
 from ngcsimlib.context import Context
 ## import model-specific elements
 from ngclearn.components.neurons.graded.rateCell import RateCell
@@ -40,11 +40,11 @@ with Context("Model") as model:  ## model/simulation definition
                     prior=("gaussian", gamma), integration_type="euler", key=subkeys[0])
 
     ## instantiate desired core commands that drive the simulation
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> cell.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> cell.reset)
     model.wrap_and_add_command(jit(reset_process.pure), name="reset")
 

diff --git a/docs/tutorials/neurocog/short_term_plasticity.md b/docs/tutorials/neurocog/short_term_plasticity.md
@@ -60,7 +60,7 @@ STF-dominated dynamics):
 ```python 
 from jax import numpy as jnp, random, jit
 from ngcsimlib.context import Context
-from ngcsimlib.compilers.process import Process
+from ngclearn.utils import JaxProcess
 ## import model-specific mechanisms
 from ngclearn.components import PoissonCell, STPDenseSynapse, LIFCell
 import ngclearn.utils.weight_distribution as dist
@@ -98,13 +98,13 @@ with Context("Model") as model:
     W.inputs << z0.outputs ## z0 -> W
     z1.j << W.outputs ## W -> z1
 
-    advance_process = (Process()
+    advance_process = (JaxProcess()
                        >> z0.advance_state
                        >> W.advance_state
                        >> z1.advance_state)
     model.wrap_and_add_command(jit(advance_process.pure), name="advance")
 
-    reset_process = (Process()
+    reset_process = (JaxProcess()
                      >> z0.reset
                      >> z1.reset
                      >> W.reset)