ngclearn.components package

Subpackages

Submodules

ngclearn.components.jaxComponent module

class ngclearn.components.jaxComponent.JaxComponent(*args, **kwargs)[source]

Bases: Component

Base Jax component that all Jax-based cells and synapses inherit from.

Parameters:
  • name – the string name of this cell

  • key – PRNG key to control determinism of any underlying random values associated with this cell

load(directory: str)[source]

The default load method for JaxComponents, it is expected to work with the default save. If the save method is modified this one will need to be modified too.

Parameters:

directory – The directory to load the .npz file.

save(directory: str)[source]

The default save method for JaxComponents, it stores the values of all non-targeted (non-wired) compartments into a .npz file.

Parameters:

directory – The directory to save the .npz file.

Module contents