skainet-lang-core/sk.ainet.lang.nn.topology

Package-level declarations

Types

class MLP<T : DType, V>(modules: Module<T, V>, val name: String = "FeedForwardNetwork") : Module<T, V> , ModuleParameters<T, V>

interface ModuleNode

A generic, dtype-agnostic node interface for traversing module hierarchies. This allows programmatic traversal, parameter collection, and introspection without depending on the input/output dtype generics of Module.

ModuleNodeAdapter

abstract class ModuleNodeAdapter : ModuleNode

Default ModuleNode implementation adapter for legacy Module types. This provides a lightweight bridge to the traversal API without changing the existing Module APIs beyond implementing ModuleNode.

ModuleParameter

sealed class ModuleParameter<T : DType, V>

ModuleParameters

interface ModuleParameters<T : DType, V>

data class Parameter(val name: String, val moduleParameter: ModuleParameter<*, *>)

Lightweight wrapper for a trainable parameter. Used for optimizer registration and tracking.

Functions

fun <T : DType, V> List<ModuleParameter<T, V>>.bias(): ModuleParameter.BiasParameter<T, V>

fun bindPaths(root: ModuleNode, base: String = root.name, separator: String = "/")

fun <T : DType, V> List<ModuleParameter<T, V>>.by(name: String): ModuleParameter<T, V>?

fun ModuleNode.collectParams(): List<ModuleParameter<*, *>>

fun ModuleNode.findById(id: String): ModuleNode?

fun ModuleNode.findByPath(path: String): ModuleNode?

fun ModuleNode.walkDepthFirst(visit: (ModuleNode) -> Unit)

Traversal utilities for ModuleNode hierarchies.

fun <T : DType, V> List<ModuleParameter<T, V>>.weights(): ModuleParameter.WeightParameter<T, V>

fun Iterable<ModuleParameter<*, *>>.zeroGrad()

Zero accumulated gradients for a collection of module parameters.