Uses of Interface
io.github.kirstenali.deepj.layers.Layer
Packages that use Layer
Package
Description
Neural networks composed from DeepJ tensors, layers, autograd and backend operations.
-
Uses of Layer in io.github.kirstenali.deepj.layers
Subinterfaces of Layer in io.github.kirstenali.deepj.layersClasses in io.github.kirstenali.deepj.layers that implement LayerModifier and TypeClassDescriptionfinal classfinal classfinal classPredicts tokens using the same parameter that supplies the input embeddings. -
Uses of Layer in io.github.kirstenali.deepj.layers.transformer
Classes in io.github.kirstenali.deepj.layers.transformer that implement Layer -
Uses of Layer in io.github.kirstenali.deepj.layers.transformer.attention
Classes in io.github.kirstenali.deepj.layers.transformer.attention that implement LayerModifier and TypeClassDescriptionfinal classfinal classclassfinal class -
Uses of Layer in io.github.kirstenali.deepj.layers.transformer.blocks
Classes in io.github.kirstenali.deepj.layers.transformer.blocks that implement LayerModifier and TypeClassDescriptionfinal classfinal classclassfinal class -
Uses of Layer in io.github.kirstenali.deepj.layers.transformer.norm
Subinterfaces of Layer in io.github.kirstenali.deepj.layers.transformer.normClasses in io.github.kirstenali.deepj.layers.transformer.norm that implement Layer -
Uses of Layer in io.github.kirstenali.deepj.nn
Classes in io.github.kirstenali.deepj.nn that implement LayerModifier 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.Methods in io.github.kirstenali.deepj.nn that return types with arguments of type LayerModifier and TypeMethodDescriptionAutoencoder.children()Conv1D.children()EncoderAttention.children()Module.children()Pool1D.children()Recurrent.children()RecurrentCell.children()Sequential.children()TransformerEncoder.children()TransformerEncoderLayer.children()Sequential.layers()Methods in io.github.kirstenali.deepj.nn with parameters of type LayerModifier and TypeMethodDescriptionstatic voidstatic voidstatic TrainerNdSupervisedTraining.trainer(Layer model, LossFunction loss, ParameterOptimizer optimizer, Tensor inputs, Tensor targets, long seed) Constructors in io.github.kirstenali.deepj.nn with parameters of type LayerModifierConstructorDescriptionAutoencoder(Layer encoder, Layer decoder) Sequential(Layer... layers) -
Uses of Layer in io.github.kirstenali.deepj.training
Methods in io.github.kirstenali.deepj.training with parameters of type LayerModifier and TypeMethodDescriptionstatic TrainerSupervisedTraining.trainer(Layer model, LossFunction lossFn, ParameterOptimizer opt, Tensor xAll, Tensor yAll, long seed) -
Uses of Layer in io.github.kirstenali.deepj.transformer
Classes in io.github.kirstenali.deepj.transformer that implement LayerModifier and TypeClassDescriptionfinal classRepeats shared grouped-query attention and SwiGLU blocks; each pass saves its own activations.final recordMethods in io.github.kirstenali.deepj.transformer that return types with arguments of type LayerModifier and TypeMethodDescriptionTransformerStack.blocks()Returns the value of theblocksrecord component.Constructor parameters in io.github.kirstenali.deepj.transformer with type arguments of type LayerModifierConstructorDescriptionTransformerStack(List<Layer> blocks) Creates an instance of aTransformerStackrecord class.