mlax

• mlax package
  • Subpackages
    • mlax.nn package
      • Submodules
      • mlax.nn.bias module
      • mlax.nn.conv module
      • mlax.nn.embed module
      • mlax.nn.f module
      • mlax.nn.functional module
      • mlax.nn.linear module
      • mlax.nn.parallel module
      • mlax.nn.scaler module
      • mlax.nn.series module
      • mlax.nn.z_norm module
      • Module contents
  • Submodules
  • mlax.module module
    • Module
      • Module.apply()
      • Module.combine()
      • Module.filter()
      • Module.filter_with_path()
      • Module.forward()
      • Module.partition()
      • Module.partition_with_path()
      • Module.setup()
      • Module.tree_flatten_with_keys()
      • Module.tree_unflatten()
    • Parameter
      • Parameter.tree_flatten()
      • Parameter.tree_unflatten()
    • is_leaf_param()
    • is_non_trainable_param()
    • is_trainable_param()
  • Module contents