pcax package

Subpackages

• pcax.core package
  • Module contents
    • BaseModule
      • BaseModule.submodules()
    • BaseParam
      • BaseParam.__init__()
      • BaseParam.get()
      • BaseParam.set()
    • Module
      • Module.MODE
      • Module.eval()
      • Module.is_eval
      • Module.is_train
      • Module.mode()
      • Module.train()
    • Param
      • Param.dtype
      • Param.get()
      • Param.ndim
      • Param.set()
      • Param.shape
    • ParamCache
    • ParamDict
      • ParamDict.get()
      • ParamDict.set()
    • RandomKeyGenerator
      • RandomKeyGenerator.__call__()
      • RandomKeyGenerator.__init__()
      • RandomKeyGenerator.seed()
    • get()
    • set()
    • static()
    • tree_apply()
    • tree_extract()
    • tree_inject()
    • tree_ref()
    • tree_unref()
• pcax.functional package
  • Module contents
    • _BaseTransform
      • _BaseTransform.__call__()
      • _BaseTransform.__init__()
      • _BaseTransform._process_mask()
    • cond()
    • jit()
    • scan()
    • switch()
    • value_and_grad()
    • vmap()
    • while_loop()
• pcax.nn package
  • Module contents
    • AvgPool2d
    • Conv
    • Conv2d
    • Layer
    • LayerNorm
    • LayerParam
    • LayerState
    • Linear
    • MaxPool2d
    • shared()
• pcax.predictive_coding package
  • Module contents
    • EnergyModule
      • EnergyModule.__init__()
      • EnergyModule.clear_params()
      • EnergyModule.energy()
      • EnergyModule.status
    • Ruleset
      • Ruleset.__init__()
      • Ruleset.apply_get_transformation()
      • Ruleset.apply_set_transformation()
      • Ruleset.filter()
    • STATUS
      • STATUS.ALL
      • STATUS.INIT
      • STATUS.NONE
    • Vode
      • Vode.__call__()
      • Vode.__init__()
      • Vode.energy()
      • Vode.get()
      • Vode.set()
    • VodeParam
      • VodeParam.Cache
    • ce_energy()
    • se_energy()
    • zero_energy()
• pcax.utils package
  • Module contents
    • Mask
      • Mask.__call__()
      • Mask.__init__()
      • Mask.apply()
    • Optim
      • Optim.__init__()
      • Optim.apply_updates()
      • Optim.clear()
      • Optim.init()
      • Optim.step()
    • load_params()
    • m
      • m.has()
      • m.has_not()
    • save_params()
    • step()

Module contents