Uses of Interface
io.github.kirstenali.deepj.training.Trainable
Packages that use Trainable
Package
Description
Neural networks composed from DeepJ tensors, layers, autograd and backend operations.
-
Uses of Trainable in io.github.kirstenali.deepj.layers
Subinterfaces of Trainable in io.github.kirstenali.deepj.layersClasses in io.github.kirstenali.deepj.layers that implement TrainableModifier and TypeClassDescriptionfinal classfinal classfinal classPredicts tokens using the same parameter that supplies the input embeddings. -
Uses of Trainable in io.github.kirstenali.deepj.layers.transformer
Classes in io.github.kirstenali.deepj.layers.transformer that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.layers.transformer.attention
Classes in io.github.kirstenali.deepj.layers.transformer.attention that implement TrainableModifier and TypeClassDescriptionfinal classfinal classclassfinal class -
Uses of Trainable in io.github.kirstenali.deepj.layers.transformer.blocks
Classes in io.github.kirstenali.deepj.layers.transformer.blocks that implement TrainableModifier and TypeClassDescriptionfinal classfinal classclassfinal class -
Uses of Trainable in io.github.kirstenali.deepj.layers.transformer.norm
Subinterfaces of Trainable in io.github.kirstenali.deepj.layers.transformer.normClasses in io.github.kirstenali.deepj.layers.transformer.norm that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.nn
Classes in io.github.kirstenali.deepj.nn that implement TrainableModifier and TypeClassDescriptionfinal classAdapts DeepJ activations to the existing Layer contract.final classComposable encoder/decoder, sharing the same parameter and autograd system as every DeepJ layer.final classChannel-axis (axis 1) normalization, using backend reductions and persistent FLOAT32 running statistics.final classTemporal cross-correlation for [batch, channels, length], sharing the 2D implementation.final classCross-correlation of [batch, channels, height, width] inputs, using gather and matrix multiplication.final classInverted dropout.final classBidirectional multi-head attention with optional padding masks and backend-resident softmax.final classFlattens an inclusive axis range; defaults to preserving the first (batch) axis.final classfinal classAverages every spatial axis of [batch, channels, ...], returning [batch, channels].final classBatch-first GRU layer; compose layers with Sequential for stacked recurrence.final classReset-after GRU, with reset/update/candidate gate order and separate projection biases.final classBatch-first LSTM layer; compose layers with Sequential for stacked recurrence.final classLSTM with input/forget/candidate/output gate order and explicit cell memory.classAdditive layer base with recursive training mode and identity-deduplicated parameters.final classMax/average pooling of [batch, channels, length] inputs.final classFloor-mode max/average pooling of channel-first images.classBatch-first recurrent sequence execution, with optional lengths and caller-owned initial/final state.classShared input/state validation and DeepJ linear projections for recurrent cells.final classReshapes the full tensor, with at most one inferred (-1) dimension.final classBatch-first RNN layer; compose layers with Sequential for stacked recurrence.final classElman recurrent cell with tanh activation.final classOrdered composition of existing or new DeepJ layers.final classStacked encoder for [batch, time, width] embeddings, with padding masks and masked mean pooling.final classPre-normalized, bidirectional encoder block reusing DeepJ linear layers and LayerNorm. -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer
Subinterfaces of Trainable in io.github.kirstenali.deepj.nn.transformerClasses in io.github.kirstenali.deepj.nn.transformer that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer.loop
Classes in io.github.kirstenali.deepj.nn.transformer.loop that implement TrainableModifier and TypeClassDescriptionfinal classA looped causal decoder reusing grouped-query attention and SwiGLU blocks across passes. -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer.nova
Classes in io.github.kirstenali.deepj.nn.transformer.nova that implement TrainableModifier and TypeClassDescriptionfinal classA decoder with grouped-query attention, optional QK normalization, and tied embeddings. -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer.orbit
Classes in io.github.kirstenali.deepj.nn.transformer.orbit that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer.origin
Classes in io.github.kirstenali.deepj.nn.transformer.origin that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.nn.transformer.prism
Classes in io.github.kirstenali.deepj.nn.transformer.prism that implement Trainable -
Uses of Trainable in io.github.kirstenali.deepj.transformer
Classes in io.github.kirstenali.deepj.transformer that implement TrainableModifier and TypeClassDescriptionfinal classRepeats shared grouped-query attention and SwiGLU blocks; each pass saves its own activations.final record -
Uses of Trainable in io.github.kirstenali.deepj.transformer.embeddings
Classes in io.github.kirstenali.deepj.transformer.embeddings that implement Trainable