Index

A B C D E F G H I K L M N O P Q R S T U V W X Y Z
All Classes and Interfaces|All Packages|Constant Field Values

A

accumulatedOutput() - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Returns the value of the accumulatedOutput record component.

accumulationType() - Method in enum class io.github.kirstenali.deepj.tensor.DType

Activation - Class in io.github.kirstenali.deepj.nn

Adapts DeepJ activations to the existing Layer contract.

Activation(ActivationFunction) - Constructor for class io.github.kirstenali.deepj.nn.Activation

ActivationFunction - Interface in io.github.kirstenali.deepj.activations

ActivationOps - Class in io.github.kirstenali.deepj.autograd

AdamW - Class in io.github.kirstenali.deepj.optimisers

AdamW(float, float, float, float, float) - Constructor for class io.github.kirstenali.deepj.optimisers.AdamW

AdamW.State - Record Class in io.github.kirstenali.deepj.optimisers

adamWUpdate(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

adamWUpdate(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

adamWUpdate(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

adamWUpdate(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

add(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

add(Tensor, int, int[], Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdIndexing

add(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BinaryOps

add(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

add(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

add(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

add(Tensor, Tensor, boolean) - Static method in class io.github.kirstenali.deepj.autograd.BroadcastOps

addBroadcastCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addBroadcastCols(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

addBroadcastRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addBroadcastRows(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addBroadcastRows(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addBroadcastRows(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

addInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addInPlace(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

addMatrixRows(Tensor, int[], Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdIndexing

addRowVector(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addRowVector(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addRowVector(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addRowVector(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

addScalar(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addScalar(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

addScalarInPlace(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

addScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

addScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

addScalarInPlace(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

affine(Tensor, Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.layers.ProjectionOps

allocatedOnGpu - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

allocBuffers(int[], int[], int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

allocTypedBuffers(int[], int[], DType[], int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

Physical element counts and dtypes, independent of logical Tensor shape.

alpacaSplit() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns the value of the alpacaSplit record component.

alternatives(int) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

apply(float) - Method in interface io.github.kirstenali.deepj.tensor.CpuNdElementwise.Unary

apply(float, float) - Method in interface io.github.kirstenali.deepj.tensor.CpuNdElementwise.Binary

apply(Tensor) - Method in interface io.github.kirstenali.deepj.autograd.BackwardRule

apply(TensorBackend, Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.tensor.CrossEntropyDefaults

apply(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdSoftmax

apply(Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdCrossEntropy

apply(Tensor, int, int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

apply(Tensor, int, int, int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

apply(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdAssignment

apply(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdMatmul

apply(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdSwiGlu

apply(Tensor, Tensor, Tensor, float, boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdNormalization

apply(String) - Method in enum class io.github.kirstenali.deepj.nn.transformer.PromptFormat

applyBackward(Tensor, int, int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

applyInference(Tensor, int, int, int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

Continuing positions for rolling inference; persistent rotary tables never grow.

applyInitScale(float) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

architecture() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns the value of the architecture record component.

Architecture(int, int, int, int, int, int, int, float, float) - Constructor for record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Creates an instance of a Architecture record class.

area() - Method in record class io.github.kirstenali.deepj.nn.Window2D

arithmetic() - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Returns the value of the arithmetic record component.

assign(Tensor, int, Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.CpuRowAccess

assignToDevice(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

attend(Object, Tensor, Tensor, Tensor, int, int, float) - Method in class io.github.kirstenali.deepj.inference.InferenceContext

AttentionGradients - Record Class in io.github.kirstenali.deepj.tensor

AttentionGradients(Tensor, Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.AttentionGradients

Creates an instance of a AttentionGradients record class.

AttentionOps - Class in io.github.kirstenali.deepj.autograd

Autoencoder - Class in io.github.kirstenali.deepj.nn

Composable encoder/decoder, sharing the same parameter and autograd system as every DeepJ layer.

Autoencoder(Layer, Layer) - Constructor for class io.github.kirstenali.deepj.nn.Autoencoder

Autograd - Class in io.github.kirstenali.deepj.autograd

average(int) - Static method in class io.github.kirstenali.deepj.nn.Pool1D

average(int) - Static method in class io.github.kirstenali.deepj.nn.Pool2D

AVERAGE - Enum constant in enum class io.github.kirstenali.deepj.nn.Pool2D.Reduction

AxisReductionOps - Class in io.github.kirstenali.deepj.autograd

B

backbone() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Dimensions for the grouped-query attention and bias-free SwiGLU blocks also used by Nova.

backend() - Static method in class io.github.kirstenali.deepj.tensor.Tensor

backward() - Method in class io.github.kirstenali.deepj.tensor.Tensor

backward(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.Autograd

backward(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

backward(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.Autograd

backward(Tensor, Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdSwiGlu

backward(Tensor, Tensor, Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

backward(Tensor, Tensor, Tensor, Tensor, boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdNormalization

BackwardBuffers - Class in io.github.kirstenali.deepj.tensor

Internal lifetime tracking for non-escaping buffers produced by library backward rules.

BackwardRule - Interface in io.github.kirstenali.deepj.autograd

A first-order vector-Jacobian product, in the same order as the operation's inputs.

BaseChatApp - Class in io.github.kirstenali.deepj.chatui

BaseChatApp() - Constructor for class io.github.kirstenali.deepj.chatui.BaseChatApp

Batch - Record Class in io.github.kirstenali.deepj.data

Batch(int[][], int[][]) - Constructor for record class io.github.kirstenali.deepj.data.Batch

Batch(int[][], int[][], boolean[][]) - Constructor for record class io.github.kirstenali.deepj.data.Batch

Creates an instance of a Batch record class.

batchedMatmul(Tensor, Tensor, int, boolean, boolean) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

batchedMatmul(Tensor, Tensor, int, boolean, boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

batchedMatmul(Tensor, Tensor, int, boolean, boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

batchedMatmul(Tensor, Tensor, int, boolean, boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

batches() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns the value of the batches record component.

batches(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.SequenceLayout

BatchNorm - Class in io.github.kirstenali.deepj.nn

Channel-axis (axis 1) normalization, using backend reductions and persistent FLOAT32 running statistics.

BatchNorm(int) - Constructor for class io.github.kirstenali.deepj.nn.BatchNorm

BatchNorm(int, float, float, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.BatchNorm

batchSize() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the batchSize record component.

batchSize() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns the value of the batchSize record component.

BatchSource - Interface in io.github.kirstenali.deepj.data

beforeMutation() - Method in class io.github.kirstenali.deepj.autograd.GradientState

begin() - Static method in class io.github.kirstenali.deepj.tensor.BackwardBuffers

begin(int) - Method in class io.github.kirstenali.deepj.inference.InferenceContext

BenchmarkDeepJ100M - Class in io.github.kirstenali.deepj.examples

Measures synthetic full training steps without reading datasets or saving checkpoints.

BenchmarkDeepJLoop100M - Class in io.github.kirstenali.deepj.examples

Synthetic full training updates; reads no dataset or checkpoint and saves nothing.

beta1() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the beta1 record component.

beta2() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the beta2 record component.

BFLOAT16 - Enum constant in enum class io.github.kirstenali.deepj.tensor.DType

bias() - Method in class io.github.kirstenali.deepj.layers.Linear

bias() - Method in class io.github.kirstenali.deepj.nn.Conv1D

bias() - Method in class io.github.kirstenali.deepj.nn.Conv2D

BiasFreeLinear - Class in io.github.kirstenali.deepj.layers

BiasFreeLinear(int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.BiasFreeLinear

BiasFreeLinear(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.BiasFreeLinear

binary(Tensor, Tensor, CpuNdElementwise.Binary) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

binaryInto(Tensor, Tensor, Tensor, CpuNdElementwise.Binary) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

BinaryOps - Class in io.github.kirstenali.deepj.autograd

bindGradient(Tensor) - Method in class io.github.kirstenali.deepj.autograd.GradientState

bindTensorToBuffer(Tensor, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

blocks() - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

Returns the value of the blocks record component.

BPEModel - Record Class in io.github.kirstenali.deepj.tokenizers.bpe

BPEModel(List<byte[]>, Map<String, Integer>, List<TokenPair>, Map<TokenPair, Integer>, int) - Constructor for record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

BPEModel(List<byte[]>, Map<String, Integer>, List<TokenPair>, Map<TokenPair, Integer>, int, int, Map<String, Integer>) - Constructor for record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Creates an instance of a BPEModel record class.

BPEModelIO - Class in io.github.kirstenali.deepj.tokenizers.bpe

BPETokenizer - Class in io.github.kirstenali.deepj.tokenizers.bpe

BPETokenizer(BPEModel) - Constructor for class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

BPETrainer - Class in io.github.kirstenali.deepj.tokenizers.bpe

BPETrainer() - Constructor for class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

BroadcastOps - Class in io.github.kirstenali.deepj.autograd

buffers() - Method in class io.github.kirstenali.deepj.nn.BatchNorm

buffers() - Method in class io.github.kirstenali.deepj.nn.Module

Persistent non-parameter tensors, in deterministic traversal order.

build() - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

build() - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

build() - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

byteLength() - Method in class io.github.kirstenali.deepj.tensor.GpuBuffer

bytes() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns the value of the bytes record component.

bytes() - Method in enum class io.github.kirstenali.deepj.tensor.DType

bytes() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

ByteTokenizer - Class in io.github.kirstenali.deepj.tokenizers

ByteTokenizer() - Constructor for class io.github.kirstenali.deepj.tokenizers.ByteTokenizer

C

capture(Supplier<T>) - Method in class io.github.kirstenali.deepj.tensor.BackwardBuffers

cast(Tensor, DType) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

cast(Tensor, DType) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

cast(Tensor, DType) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

cast(Tensor, DType) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

cast(Tensor, DType) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

castToDevice(Tensor, DType) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

causal(Tensor, Tensor, Tensor, int, float) - Static method in class io.github.kirstenali.deepj.autograd.AttentionOps

causal(Tensor, Tensor, Tensor, int, int, float) - Static method in class io.github.kirstenali.deepj.autograd.AttentionOps

causalAttention(Tensor, Tensor, Tensor, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

causalAttention(Tensor, Tensor, Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

causalAttention(Tensor, Tensor, Tensor, int, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

causalAttention(Tensor, Tensor, Tensor, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

CausalAttentionBackend - Interface in io.github.kirstenali.deepj.tensor

Optional backend capability for memory-efficient, fused causal attention.

causalAttentionBackward(Tensor, Tensor, Tensor, CausalAttentionResult, Tensor, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

causalAttentionBackward(Tensor, Tensor, Tensor, CausalAttentionResult, Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

causalAttentionBackward(Tensor, Tensor, Tensor, CausalAttentionResult, Tensor, int, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

causalAttentionBackward(Tensor, Tensor, Tensor, CausalAttentionResult, Tensor, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

CausalAttentionResult - Record Class in io.github.kirstenali.deepj.tensor

Output, row statistics and unrounded FP32 output needed by tiled backward.

CausalAttentionResult(Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

CausalAttentionResult(Tensor, Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Creates an instance of a CausalAttentionResult record class.

CausalLM - Interface in io.github.kirstenali.deepj.nn.transformer

CausalLMEvaluation - Class in io.github.kirstenali.deepj.training

CausalLMTraining - Class in io.github.kirstenali.deepj.training

causalMask(int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

causalMask(int) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

causalMask(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

causalMask(Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

causalSoftmax(Tensor, int, float) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

causalSoftmax(Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

causalSoftmax(Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

causalSoftmax(Tensor, int, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

cell() - Method in class io.github.kirstenali.deepj.nn.Recurrent

cell() - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

Returns the value of the cell record component.

ChatApp - Class in io.github.kirstenali.deepj.chatui

ChatApp() - Constructor for class io.github.kirstenali.deepj.chatui.ChatApp

ChatMessage - Record Class in io.github.kirstenali.deepj.chatui

ChatMessage(String, boolean) - Constructor for record class io.github.kirstenali.deepj.chatui.ChatMessage

Creates an instance of a ChatMessage record class.

ChatService - Interface in io.github.kirstenali.deepj.chatui

ChatWindow - Class in io.github.kirstenali.deepj.chatui

checkpointEvery() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the checkpointEvery record component.

children() - Method in class io.github.kirstenali.deepj.nn.Autoencoder

children() - Method in class io.github.kirstenali.deepj.nn.Conv1D

children() - Method in class io.github.kirstenali.deepj.nn.EncoderAttention

children() - Method in class io.github.kirstenali.deepj.nn.Module

children() - Method in class io.github.kirstenali.deepj.nn.Pool1D

children() - Method in class io.github.kirstenali.deepj.nn.Recurrent

children() - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

children() - Method in class io.github.kirstenali.deepj.nn.Sequential

children() - Method in class io.github.kirstenali.deepj.nn.TransformerEncoder

children() - Method in class io.github.kirstenali.deepj.nn.TransformerEncoderLayer

chooseModel() - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

clamp(float, float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

clamp(Tensor, float, float) - Static method in class io.github.kirstenali.deepj.autograd.SelectionOps

The derivative is zero at either clipping boundary.

clamp(Tensor, float, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

clamp(Tensor, float, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

clamp(Tensor, float, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

classes(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CrossEntropyChecks

close() - Method in class io.github.kirstenali.deepj.autograd.GradMode

close() - Method in class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

close() - Method in class io.github.kirstenali.deepj.data.RandomAccessTextDataset

close() - Method in class io.github.kirstenali.deepj.data.SequentialTextDataset

close() - Method in class io.github.kirstenali.deepj.inference.InferenceContext

close() - Method in interface io.github.kirstenali.deepj.nn.transformer.GenerationSession

close() - Method in class io.github.kirstenali.deepj.tensor.BackwardBuffers

close() - Method in class io.github.kirstenali.deepj.tensor.TensorScope

codaLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the codaLayers record component.

cols - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

cols() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the cols record component.

columns(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ReductionOps

compareTo(TokenPair) - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

completedSteps() - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Returns the value of the completedSteps record component.

ComputeGraph - Class in io.github.kirstenali.deepj.tensor

ComputeGraph(GpuRuntime) - Constructor for class io.github.kirstenali.deepj.tensor.ComputeGraph

concatenate(int, Tensor...) - Static method in class io.github.kirstenali.deepj.nn.TensorJoin

config() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

config() - Method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

config() - Method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

config() - Method in class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

config() - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

config() - Method in class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

config() - Method in class io.github.kirstenali.deepj.nn.TransformerEncoder

config() - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Returns the value of the config record component.

CONFIG_FILE - Static variable in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

ContextWindowMode - Enum Class in io.github.kirstenali.deepj.nn.transformer

Generation only: rolling mode preserves cached history rather than rebuilding each shifted window.

contiguous() - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

contiguous(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

Conv1D - Class in io.github.kirstenali.deepj.nn

Temporal cross-correlation for [batch, channels, length], sharing the 2D implementation.

Conv1D(int, int, int, int, int, int, boolean, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.Conv1D

Conv1D(int, int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.Conv1D

Conv2D - Class in io.github.kirstenali.deepj.nn

Cross-correlation of [batch, channels, height, width] inputs, using gather and matrix multiplication.

Conv2D(int, int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.Conv2D

Conv2D(int, int, Window2D, boolean, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.Conv2D

copy() - Method in class io.github.kirstenali.deepj.tensor.Tensor

Bit-preserving independent storage with an identity autograd edge.

copy(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

copy(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

copy(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Independent storage with the same float bits, without recording an autograd operation.

copy(String) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

copyFrom(Tensor) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

Independent value assignment with an explicit cast to this tensor's dtype; shapes must match.

copyInto(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

copyInto(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

copyInto(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Explicitly assigns logical values, casting to the destination's storage dtype.

copyToDevice(Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

coreLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the coreLayers record component.

coreResidualScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the coreResidualScale record component.

corpus() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Returns the value of the corpus record component.

corpusBytes() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns the value of the corpusBytes record component.

CosineLearningRateSchedule - Record Class in io.github.kirstenali.deepj.training

CosineLearningRateSchedule(float, float, int, int) - Constructor for record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Creates an instance of a CosineLearningRateSchedule record class.

count(DeepJLoopConfig) - Static method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopParameterCount

count(DeepJNovaConfig) - Static method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaParameterCount

count(DeepJPrismConfig) - Static method in class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismParameterCount

coversStorage() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Whether immutable logical geometry covers the whole allocation contiguously from offset zero.

CpuBackend - Class in io.github.kirstenali.deepj.tensor.cpu

CpuBackend() - Constructor for class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

CpuNdAdamW - Class in io.github.kirstenali.deepj.tensor

Logical-layout Adam arithmetic matches the original contiguous FLOAT32 update order.

CpuNdAssignment - Class in io.github.kirstenali.deepj.tensor

Assigns logical values with explicit dtype conversion, including overlapping source/destination views.

CpuNdCrossEntropy - Class in io.github.kirstenali.deepj.tensor

Stable final-axis loss and FLOAT32 gradient, including strided/compact logits and response masks.

CpuNdElementwise - Class in io.github.kirstenali.deepj.tensor

CPU logical-layout kernels with real compact storage and FLOAT32 arithmetic.

CpuNdElementwise.Binary - Interface in io.github.kirstenali.deepj.tensor

CpuNdElementwise.Unary - Interface in io.github.kirstenali.deepj.tensor

CpuNdIndexing - Class in io.github.kirstenali.deepj.tensor

Logical ND index selection and deterministic repeated-index accumulation.

CpuNdMatmul - Class in io.github.kirstenali.deepj.tensor

General strided/broadcast CPU matmul; the existing FLOAT32 matrix fast path remains separate.

CpuNdNormalization - Class in io.github.kirstenali.deepj.tensor

Last-axis CPU normalization, with FLOAT32 statistics and no intermediate compact rounding.

CpuNdPosition - Class in io.github.kirstenali.deepj.tensor

Typed position operations accumulate in FLOAT32 and round only the final stored activation.

CpuNdReduction - Class in io.github.kirstenali.deepj.tensor

Axis reductions accumulate in FLOAT32 and round compact output only once per result value.

CpuNdSoftmax - Class in io.github.kirstenali.deepj.tensor

Stable logical-axis softmax without an intermediate compact-dtype exponential tensor.

CpuNdStatistics - Class in io.github.kirstenali.deepj.tensor

Two-pass FLOAT32 statistics without a full-sized FLOAT32 shadow of compact input.

CpuNdSwiGlu - Class in io.github.kirstenali.deepj.tensor

Equal-shaped SwiGLU operands decode to FLOAT32 and only round the completed output.

CpuRowAccess - Class in io.github.kirstenali.deepj.tensor

Checked CPU row kernels; the Tensor facade owns autograd mutation fences as for other backends.

cpuStale - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

createChatService() - Method in class io.github.kirstenali.deepj.chatui.BaseChatApp

createChatService() - Method in class io.github.kirstenali.deepj.chatui.ChatApp

createOutputLike(GpuBuffer, Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Shares only immutable canonical geometry, never storage, gradients or caller-owned arrays.

createOutputTensor(GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

createOutputTensor(GpuBuffer, int...) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

crossEntropy(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropy(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropy(Tensor, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

crossEntropy(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropy(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropy(Tensor, int[], boolean[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

CrossEntropyChecks - Class in io.github.kirstenali.deepj.tensor

Class scores occupy the final axis; every leading position has one integer target.

CrossEntropyDefaults - Class in io.github.kirstenali.deepj.tensor

Unfused backend composition preserves the original matrix loss/gradient evaluation order.

crossEntropyGradient(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropyGradient(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropyGradient(Tensor, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

crossEntropyGradient(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropyGradient(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropyGradient(Tensor, int[], boolean[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

crossEntropyLoss(int[]) - Method in class io.github.kirstenali.deepj.tensor.Tensor

crossEntropyLoss(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropyLoss(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropyLoss(Tensor, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

crossEntropyLoss(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

crossEntropyLoss(Tensor, int[], boolean[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

crossEntropyLoss(Tensor, int[], boolean[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

CrossEntropyLoss - Class in io.github.kirstenali.deepj.loss

CrossEntropyLoss() - Constructor for class io.github.kirstenali.deepj.loss.CrossEntropyLoss

CrossEntropyResult - Record Class in io.github.kirstenali.deepj.tensor

CrossEntropyResult(Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Creates an instance of a CrossEntropyResult record class.

CURRENT_FORMAT_VERSION - Static variable in class io.github.kirstenali.deepj.persistence.ModelSerializer

CURRENT_FORMAT_VERSION - Static variable in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

D

data() - Method in class io.github.kirstenali.deepj.tensor.Tensor

Materializes and returns the backing FLOAT32 array of a full contiguous tensor.

decode(int[]) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

decode(int[]) - Method in class io.github.kirstenali.deepj.tokenizers.ByteTokenizer

decode(int[]) - Method in interface io.github.kirstenali.deepj.tokenizers.Tokenizer

decode(Tensor) - Method in class io.github.kirstenali.deepj.nn.Autoencoder

DecoderOnlyModel - Class in io.github.kirstenali.deepj.nn.transformer

DecoderOnlyModel(Embedding, TransformerStack, NormLayer, Projection) - Constructor for class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

DeepJChat - Class in io.github.kirstenali.deepj.nn.transformer

DeepJChatService - Class in io.github.kirstenali.deepj.chatui

DeepJChatService() - Constructor for class io.github.kirstenali.deepj.chatui.DeepJChatService

DeepJCli - Class in io.github.kirstenali.deepj.cli

Terminal entry point; the Java library remains usable without the CLI dependencies.

DeepJCli() - Constructor for class io.github.kirstenali.deepj.cli.DeepJCli

DeepJExecutor - Class in io.github.kirstenali.deepj.concurrent

DeepJLoop - Class in io.github.kirstenali.deepj.nn.transformer.loop

A looped causal decoder reusing grouped-query attention and SwiGLU blocks across passes.

DeepJLoop(DeepJLoopConfig, long) - Constructor for class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

DeepJLoop(DeepJLoopConfig, long, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

DeepJLoop(DeepJLoopConfig, long, DType) - Constructor for class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

DeepJLoopConfig - Record Class in io.github.kirstenali.deepj.nn.transformer.loop

Unique layer counts and the number of fully differentiated recurrent passes.

DeepJLoopConfig(int, int, int, int, int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

DeepJLoopConfig(int, int, int, int, int, int, int, int, int, int, boolean, float, float, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Creates an instance of a DeepJLoopConfig record class.

DeepJLoopParameterCount - Class in io.github.kirstenali.deepj.nn.transformer.loop

DeepJModelBundle - Class in io.github.kirstenali.deepj.publishing

DeepJNova - Class in io.github.kirstenali.deepj.nn.transformer.nova

A decoder with grouped-query attention, optional QK normalization, and tied embeddings.

DeepJNova(DeepJNovaConfig, long) - Constructor for class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

DeepJNova(DeepJNovaConfig, long, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

DeepJNova(DeepJNovaConfig, long, DType) - Constructor for class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

DeepJNovaConfig - Record Class in io.github.kirstenali.deepj.nn.transformer.nova

DeepJNovaConfig(int, int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

DeepJNovaConfig(int, int, int, int, int, int, int, boolean, float, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Creates an instance of a DeepJNovaConfig record class.

DeepJNovaParameterCount - Class in io.github.kirstenali.deepj.nn.transformer.nova

DeepJNovaTransformerBlock - Class in io.github.kirstenali.deepj.layers.transformer.blocks

DeepJNovaTransformerBlock(DeepJNovaConfig, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

DeepJNovaTransformerBlock(DeepJNovaConfig, Random, float) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

DeepJNovaTransformerBlock(DeepJNovaConfig, Random, float, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

DeepJOrbit - Class in io.github.kirstenali.deepj.nn.transformer.orbit

DeepJOrbit(DeepJOrbitConfig, long) - Constructor for class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

DeepJOrbit(DeepJOrbitConfig, long, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

DeepJOrbit(DeepJOrbitConfig, long, DType) - Constructor for class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

DeepJOrbitConfig - Record Class in io.github.kirstenali.deepj.nn.transformer.orbit

DeepJOrbitConfig(int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

DeepJOrbitConfig(int, int, int, int, int, int, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

DeepJOrbitConfig(int, int, int, int, int, int, float, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Creates an instance of a DeepJOrbitConfig record class.

DeepJOrbitTransformerBlock - Class in io.github.kirstenali.deepj.layers.transformer.blocks

DeepJOrbitTransformerBlock(int, int, int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

DeepJOrbitTransformerBlock(int, int, int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

DeepJOrbitTransformerBuilder - Class in io.github.kirstenali.deepj.transformer

DeepJOrbitTransformerBuilder() - Constructor for class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

DeepJOrigin - Class in io.github.kirstenali.deepj.nn.transformer.origin

DeepJOrigin(DeepJOriginConfig, long) - Constructor for class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

DeepJOrigin(DeepJOriginConfig, long, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

DeepJOrigin(DeepJOriginConfig, long, DType) - Constructor for class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

DeepJOriginChatService - Class in io.github.kirstenali.deepj.chatui

DeepJOriginChatService() - Constructor for class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

DeepJOriginConfig - Record Class in io.github.kirstenali.deepj.nn.transformer.origin

DeepJOriginConfig(int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

DeepJOriginConfig(int, int, int, int, int, int, float, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Creates an instance of a DeepJOriginConfig record class.

DeepJOriginTransformerBlock - Class in io.github.kirstenali.deepj.layers.transformer.blocks

DeepJOriginTransformerBlock(int, int, int, Supplier<ActivationFunction>, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

DeepJOriginTransformerBlock(int, int, int, Supplier<ActivationFunction>, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

DeepJOriginTransformerBlock(int, int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

DeepJOriginTransformerBuilder - Class in io.github.kirstenali.deepj.transformer

DeepJOriginTransformerBuilder() - Constructor for class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

DeepJPrism - Class in io.github.kirstenali.deepj.nn.transformer.prism

DeepJPrism(DeepJPrismConfig, long) - Constructor for class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

DeepJPrism(DeepJPrismConfig, long, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

DeepJPrism(DeepJPrismConfig, long, DType) - Constructor for class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

DeepJPrismChatService - Class in io.github.kirstenali.deepj.chatui

DeepJPrismChatService() - Constructor for class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

DeepJPrismConfig - Record Class in io.github.kirstenali.deepj.nn.transformer.prism

DeepJPrismConfig(int, int, int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

DeepJPrismConfig(int, int, int, int, int, int, int, int, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

DeepJPrismConfig(int, int, int, int, int, int, int, int, float, float) - Constructor for record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Creates an instance of a DeepJPrismConfig record class.

DeepJPrismParameterCount - Class in io.github.kirstenali.deepj.nn.transformer.prism

DeepJPrismTrainingConfig - Record Class in io.github.kirstenali.deepj.examples

DeepJPrismTrainingConfig(DeepJPrismTrainingConfig.FilesConfig, DeepJPrismTrainingConfig.Architecture, DeepJPrismTrainingConfig.Training, DeepJPrismTrainingConfig.TokenizerConfig, long) - Constructor for record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Creates an instance of a DeepJPrismTrainingConfig record class.

DeepJPrismTrainingConfig.Architecture - Record Class in io.github.kirstenali.deepj.examples

DeepJPrismTrainingConfig.FilesConfig - Record Class in io.github.kirstenali.deepj.examples

DeepJPrismTrainingConfig.TokenizerConfig - Record Class in io.github.kirstenali.deepj.examples

DeepJPrismTrainingConfig.Training - Record Class in io.github.kirstenali.deepj.examples

DeepJPrismTransformerBlock - Class in io.github.kirstenali.deepj.layers.transformer.blocks

DeepJPrismTransformerBlock(int, int, int, int, int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

DeepJPrismTransformerBlock(int, int, int, int, int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

DeepJPrismTransformerBuilder - Class in io.github.kirstenali.deepj.transformer

DeepJPrismTransformerBuilder() - Constructor for class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

DEFAULT_SPECIAL_TOKENS - Static variable in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

defaultAdamW(float) - Static method in class io.github.kirstenali.deepj.optimisers.AdamW

defaultDFF(int) - Static method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

denominator() - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Returns the value of the denominator record component.

detach() - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

detach() - Method in class io.github.kirstenali.deepj.tensor.Tensor

Independent detached storage, preserving DeepJ's existing copy-based detach behavior.

dFF() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the dFF record component.

dFF() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the dFF record component.

dFF() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the dFF record component.

dFF() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the dFF record component.

dFF() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the dFF record component.

dFF() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the dFF record component.

dFF() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

dFF(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

dFF(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

dFF(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

dilationHeight() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the dilationHeight record component.

dilationWidth() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the dilationWidth record component.

divide(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

divide(Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

divide(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BinaryOps

divide(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BroadcastOps

divide(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

divide(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

divide(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

divideBroadcastCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

divideBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

divideBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

divideBroadcastCols(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

divideInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

divideInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

divideInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

divideInPlace(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

divideScalar(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

divideScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

divideScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

divideScalar(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

divideScalarInPlace(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

divideScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

divideScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

divideScalarInPlace(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

dModel() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the dModel record component.

dModel() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the dModel record component.

dModel() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the dModel record component.

dModel() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the dModel record component.

dModel() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the dModel record component.

dModel() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the dModel record component.

dModel() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

dModel(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

dModel(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

dModel(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

downloadBuffer(int, float[]) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

downloadBuffer16(int, short[]) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

dropout() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns the value of the dropout record component.

Dropout - Class in io.github.kirstenali.deepj.nn

Inverted dropout.

Dropout(float, long) - Constructor for class io.github.kirstenali.deepj.nn.Dropout

Dropout(float, Random) - Constructor for class io.github.kirstenali.deepj.nn.Dropout

dtype - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

dtype() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Returns the value of the dtype record component.

dtype() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

DType - Enum Class in io.github.kirstenali.deepj.tensor

Dense floating-point storage formats.

E

elementCount() - Method in class io.github.kirstenali.deepj.tensor.GpuBuffer

elements() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

emaLoss() - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Returns the value of the emaLoss record component.

emaLoss() - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Returns the value of the emaLoss record component.

embed(int[]) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

embed(int[]) - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

embed(int[], int) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

embed(int[], int) - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

embedBatch(int[][]) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

embedBatch(int[][]) - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

Embedding - Class in io.github.kirstenali.deepj.transformer.embeddings

Embedding(int, int, Random) - Constructor for class io.github.kirstenali.deepj.transformer.embeddings.Embedding

Embedding(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.transformer.embeddings.Embedding

embeddingParameters() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

embeddingParameters() - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

encode(Tensor) - Method in class io.github.kirstenali.deepj.nn.Autoencoder

encode(String) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

encode(String) - Method in class io.github.kirstenali.deepj.tokenizers.ByteTokenizer

encode(String) - Method in interface io.github.kirstenali.deepj.tokenizers.Tokenizer

EncoderAttention - Class in io.github.kirstenali.deepj.nn

Bidirectional multi-head attention with optional padding masks and backend-resident softmax.

EncoderAttention(int, int, float, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.EncoderAttention

endExclusive() - Method in record class io.github.kirstenali.deepj.data.TextFileRange

Returns the value of the endExclusive record component.

endOfWordId() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the endOfWordId record component.

ensureGpuBuffer(Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

ensureStorageBuffer(Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Physical shared storage only; logical ND kernels must supply their own layouts.

entire(Path) - Static method in record class io.github.kirstenali.deepj.data.TextFileRange

epsilon() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the epsilon record component.

equals(Object) - Method in record class io.github.kirstenali.deepj.chatui.ChatMessage

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.data.Batch

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.data.TextFileRange

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.RecurrentResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.nn.Window2D

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Indicates whether some other object is "equal to" this one.

equals(Object) - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

Indicates whether some other object is "equal to" this one.

escape(Tensor) - Method in class io.github.kirstenali.deepj.tensor.TensorScope

eval() - Method in class io.github.kirstenali.deepj.nn.Module

evaluate(CausalLM, BatchSource, int, int) - Static method in class io.github.kirstenali.deepj.training.CausalLMEvaluation

evaluate(CausalLM, Path, Tokenizer, FixedTextValidation.Settings) - Static method in class io.github.kirstenali.deepj.training.FixedTextValidation

evaluate(CausalLM, Path, Tokenizer, FixedTextValidation.Settings, TextFileRange) - Static method in class io.github.kirstenali.deepj.training.FixedTextValidation

EvaluateDeepJ100M - Class in io.github.kirstenali.deepj.examples

Evaluates Nova on the same held-out FineWeb-Edu range used for Frisky.

EvaluateDeepJ90M - Class in io.github.kirstenali.deepj.examples

EvaluateDeepJ90MPretraining - Class in io.github.kirstenali.deepj.examples

EvaluateDeepJLoop100M - Class in io.github.kirstenali.deepj.examples

Evaluates the looped decoder on the unchanged held-out FineWeb-Edu split.

EvaluateDeepJPrismKnowledge - Class in io.github.kirstenali.deepj.examples

EvaluationResult - Record Class in io.github.kirstenali.deepj.training

EvaluationResult(double, double, long) - Constructor for record class io.github.kirstenali.deepj.training.EvaluationResult

Creates an instance of a EvaluationResult record class.

Example(String, String, int[], int) - Constructor for record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Creates an instance of a Example record class.

examples() - Method in class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset

executedLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

exp() - Method in class io.github.kirstenali.deepj.tensor.Tensor

exp(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

exp(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

exp(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

exp(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

expand(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

expand(Tensor, int...) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

expand(Tensor, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

expInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

expInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

expInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

expInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

export(Path, DeepJOrigin, BPEModel, ModelCard) - Static method in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

export(Path, DeepJPrism, BPEModel, ModelCard) - Static method in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

ExportFrisky90M - Class in io.github.kirstenali.deepj.examples

expose(Tensor) - Method in class io.github.kirstenali.deepj.tensor.BackwardBuffers

external(Supplier<T>) - Static method in class io.github.kirstenali.deepj.tensor.TensorScope

User callbacks must not inherit the library's private lifetime boundary.

F

facts() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns the value of the facts record component.

factSplit() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns the value of the factSplit record component.

ffnActivation(Supplier<ActivationFunction>) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

files() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns the value of the files record component.

FilesConfig(Path, Path, Path) - Constructor for record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Creates an instance of a FilesConfig record class.

FineTuneDeepJ100M - Class in io.github.kirstenali.deepj.examples

Fine-tunes Nova on Frisky's unchanged response-only dataset.

FineTuneDeepJ90M - Class in io.github.kirstenali.deepj.examples

FineTuneDeepJLoop100M - Class in io.github.kirstenali.deepj.examples

Runs the fine_tune stage using the existing Frisky dataset.

FineTuneDeepJPrismKnowledge - Class in io.github.kirstenali.deepj.examples

finish() - Method in class io.github.kirstenali.deepj.inference.InferenceContext

firstMoments() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the firstMoments record component.

FixedTextValidation - Class in io.github.kirstenali.deepj.training

FixedTextValidation.Settings - Record Class in io.github.kirstenali.deepj.training

Flatten - Class in io.github.kirstenali.deepj.nn

Flattens an inclusive axis range; defaults to preserving the first (batch) axis.

Flatten() - Constructor for class io.github.kirstenali.deepj.nn.Flatten

Flatten(int, int) - Constructor for class io.github.kirstenali.deepj.nn.Flatten

FLOAT16 - Enum constant in enum class io.github.kirstenali.deepj.tensor.DType

float32() - Static method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

FLOAT32 - Enum constant in enum class io.github.kirstenali.deepj.tensor.DType

floatCount() - Method in class io.github.kirstenali.deepj.tensor.GpuBuffer

flush() - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

flushOps(int[], int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

flushOps(int[], int, long[]) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

FNN - Class in io.github.kirstenali.deepj.nn

FNN(int, int[], int, Supplier<ActivationFunction>, ActivationFunction, Random) - Constructor for class io.github.kirstenali.deepj.nn.FNN

FNN(int, int[], int, Supplier<ActivationFunction>, ActivationFunction, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.FNN

FNN(int, int[], int, Supplier<ActivationFunction>, Random) - Constructor for class io.github.kirstenali.deepj.nn.FNN

forLoading() - Method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

Zero-filled random-weight slots for a complete checkpoint replacement, without device initialization.

format(Path, Path) - Static method in class io.github.kirstenali.deepj.examples.FormatAlpacaText

FormatAlpacaText - Class in io.github.kirstenali.deepj.examples

FormatAlpacaText.FormatResult - Record Class in io.github.kirstenali.deepj.examples

FormatResult(long, long) - Constructor for record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Creates an instance of a FormatResult record class.

formatted() - Method in record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Returns the value of the formatted record component.

formatted() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns the value of the formatted record component.

formatVersion(Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

forRange(int, int, IntConsumer) - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

forward(int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.PositionalEmbedding

forward(int[]) - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

forward(int[]) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

forward(int[]) - Method in class io.github.kirstenali.deepj.transformer.embeddings.Embedding

forward(int, int) - Method in class io.github.kirstenali.deepj.transformer.embeddings.PositionalEmbedding

forward(Tensor) - Method in interface io.github.kirstenali.deepj.activations.ActivationFunction

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.GELU

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.ReLU

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.Sigmoid

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.SiLU

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.Softmax

forward(Tensor) - Method in class io.github.kirstenali.deepj.activations.Tanh

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.BiasFreeLinear

forward(Tensor) - Method in interface io.github.kirstenali.deepj.layers.Layer

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.Linear

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.TiedEmbeddingProjection

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.norm.LayerNorm1D

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.norm.RMSNorm1D

forward(Tensor) - Method in class io.github.kirstenali.deepj.layers.transformer.SwiGLULayer

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Activation

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Autoencoder

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.BatchNorm

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Conv1D

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Conv2D

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Dropout

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.EncoderAttention

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Flatten

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.FNN

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.GlobalAveragePool

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Pool1D

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Pool2D

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Recurrent

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Reshape

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.Sequential

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.TransformerEncoder

forward(Tensor) - Method in class io.github.kirstenali.deepj.nn.TransformerEncoderLayer

forward(Tensor) - Method in class io.github.kirstenali.deepj.transformer.LoopedTransformer

forward(Tensor) - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

forward(Tensor, boolean[][]) - Method in class io.github.kirstenali.deepj.nn.EncoderAttention

forward(Tensor, boolean[][]) - Method in class io.github.kirstenali.deepj.nn.TransformerEncoder

forward(Tensor, boolean[][]) - Method in class io.github.kirstenali.deepj.nn.TransformerEncoderLayer

forward(Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

forward(Tensor, InferenceContext) - Method in interface io.github.kirstenali.deepj.layers.Layer

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

forward(Tensor, InferenceContext) - Method in class io.github.kirstenali.deepj.transformer.LoopedTransformer

forward(Tensor, InferenceContext) - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

forward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

forward(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.loss.LossFunction

forward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.MSELoss

forward(Tensor, Tensor, Tensor, float, boolean) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

forwardBatch(int[][]) - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

forwardBatch(int[][]) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

forwardBatch(int[][]) - Method in class io.github.kirstenali.deepj.transformer.embeddings.Embedding

forwardCausal(Tensor, int, float) - Method in class io.github.kirstenali.deepj.activations.Softmax

from(float[]) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

from(float[], int...) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

from(float[], DType) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

from(float[], DType, int...) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

from2D(float[][]) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

fromFile(Path, Tokenizer, int, long) - Static method in class io.github.kirstenali.deepj.data.TextDataset

fromIntColumn(int[]) - Static method in class io.github.kirstenali.deepj.tensor.TensorAdapters

fromIntTargets(int[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

fromStorageCode(int) - Static method in enum class io.github.kirstenali.deepj.tensor.DType

G

gate(Tensor, int) - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

gateGradient() - Method in record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Returns the value of the gateGradient record component.

gather(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

gelu() - Static method in class io.github.kirstenali.deepj.nn.Activation

gelu(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

gelu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

gelu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

gelu(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

GELU - Class in io.github.kirstenali.deepj.activations

GELU() - Constructor for class io.github.kirstenali.deepj.activations.GELU

geluActivation() - Method in class io.github.kirstenali.deepj.tensor.Tensor

geluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

geluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

geluBackward(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

geluInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

geluInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

geluInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

geluInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

generate(CausalLM, Tokenizer, TransformerConfig, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(CausalLM, Tokenizer, TransformerConfig, String, int, float, int, long, float) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(DeepJOrbit, Tokenizer, TransformerConfig, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(DeepJOrigin, Tokenizer, DeepJOriginConfig, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(DeepJPrism, Tokenizer, TransformerConfig, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(String) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

generate(String, int, float, int, long) - Method in interface io.github.kirstenali.deepj.chatui.ChatService

generate(String, int, float, int, long) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

generate(String, int, float, int, long, float) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

generate(String, int, float, int, long, float, ContextWindowMode, Consumer<String>) - Method in interface io.github.kirstenali.deepj.chatui.ChatService

generate(String, int, float, int, long, float, ContextWindowMode, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

generate(String, int, float, int, long, float, ContextWindowMode, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

generate(String, int, float, int, long, float, Consumer<String>) - Method in interface io.github.kirstenali.deepj.chatui.ChatService

generate(String, int, float, int, long, float, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

generate(String, int, float, int, long, float, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

generate(String, int, float, int, long, float, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

generate(String, int, float, int, long, Consumer<String>) - Method in interface io.github.kirstenali.deepj.chatui.ChatService

generate(String, int, float, int, long, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

generate(String, int, float, int, long, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

generate(String, int, float, int, long, Consumer<String>) - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

generate(Path, int) - Static method in class io.github.kirstenali.deepj.examples.GenerateFactDataset

generate(Function<int[], Tensor>, int, Tokenizer, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(Function<int[], Tensor>, int, Tokenizer, String, int, float, int, long, float) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

generate(Function<int[], Tensor>, TransformerConfig, Tokenizer, String, int, float, int, long) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

GenerateFactDataset - Class in io.github.kirstenali.deepj.examples

GenerateFactDataset.GenerationResult - Record Class in io.github.kirstenali.deepj.examples

GenerationResult(long, long) - Constructor for record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Creates an instance of a GenerationResult record class.

GenerationSession - Interface in io.github.kirstenali.deepj.nn.transformer

One generation request.

get(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

get(int, int) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

get(Tensor, int, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

getAppTitle() - Method in class io.github.kirstenali.deepj.chatui.BaseChatApp

getGpuTag() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

getLoadedModelName() - Method in interface io.github.kirstenali.deepj.chatui.ChatService

getLoadedModelName() - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

getLoadedModelName() - Method in class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

getLoadedModelName() - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

getNumThreads() - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

getRow(int) - Method in class io.github.kirstenali.deepj.tensor.Tensor

getRow(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

GlobalAveragePool - Class in io.github.kirstenali.deepj.nn

Averages every spatial axis of [batch, channels, ...], returning [batch, channels].

GlobalAveragePool() - Constructor for class io.github.kirstenali.deepj.nn.GlobalAveragePool

GpuAllocations - Class in io.github.kirstenali.deepj.tensor

Validates physical allocations without flattening logical tensor dimensions.

GpuBuffer - Class in io.github.kirstenali.deepj.tensor

GpuBuffer(int, int, int, boolean) - Constructor for class io.github.kirstenali.deepj.tensor.GpuBuffer

GpuBuffer(int, int, int, DType, boolean) - Constructor for class io.github.kirstenali.deepj.tensor.GpuBuffer

GpuRuntime - Interface in io.github.kirstenali.deepj.tensor

grad() - Method in class io.github.kirstenali.deepj.optimisers.Parameter

grad() - Method in class io.github.kirstenali.deepj.tensor.Tensor

gradClipNorm() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

gradClipNorm() - Method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

gradClipNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

gradClipNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

gradClipNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

gradClipNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

gradClipNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

gradient() - Method in class io.github.kirstenali.deepj.autograd.GradientState

gradient() - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Returns the value of the gradient record component.

gradient(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

gradient(Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

gradient(Tensor, int[], int, int[]) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdIndexing

gradient(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

gradient(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.loss.LossFunction

gradient(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.MSELoss

gradientState() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

GradientState - Class in io.github.kirstenali.deepj.autograd

Runtime-only state; never serialized with a tensor or checkpoint.

GradientState() - Constructor for class io.github.kirstenali.deepj.autograd.GradientState

gradientType() - Method in enum class io.github.kirstenali.deepj.tensor.DType

GradMode - Class in io.github.kirstenali.deepj.autograd

Thread-local, nestable scope for computations that must not build a backward graph.

GraphOpHistogram - Class in io.github.kirstenali.deepj.tensor

Opt-in opcode counter for the executed GPU command stream (-Ddeepj.graph.histogram=true).

GroupedQueryAttention - Class in io.github.kirstenali.deepj.layers.transformer.attention

GroupedQueryAttention(DeepJNovaConfig, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

GroupedQueryAttention(DeepJNovaConfig, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

GRU - Class in io.github.kirstenali.deepj.nn

Batch-first GRU layer; compose layers with Sequential for stacked recurrence.

GRU(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.GRU

GRU(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.GRU

GRUCell - Class in io.github.kirstenali.deepj.nn

Reset-after GRU, with reset/update/candidate gate order and separate projection biases.

GRUCell(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.GRUCell

GRUCell(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.GRUCell

H

hasCellMemory() - Method in class io.github.kirstenali.deepj.nn.LSTMCell

hasCellMemory() - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

hashCode() - Method in record class io.github.kirstenali.deepj.chatui.ChatMessage

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.data.Batch

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.data.TextFileRange

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.RecurrentResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Returns a hash code value for this object.

hashCode() - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

Returns a hash code value for this object.

hasHostStorage() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

False for lazy storage that has never been touched on the host; such storage reads as zeros.

hasSameShape(TensorStorageAccess) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

hasShape(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Tests exact logical dimensions without copying shape metadata or reading tensor values.

hasSpecialTokens() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

HeadCacheOps - Class in io.github.kirstenali.deepj.tensor

Inference-only storage operations for head-major key/value caches.

headDim - Variable in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

headDim() - Method in class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

HeadLayout - Class in io.github.kirstenali.deepj.tensor

Batch and head axes are distinct; packed matrices are temporary native-kernel views only.

headPrefixMatmul(Tensor, Tensor, int, int, boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

headPrefixMatmul(Tensor, Tensor, int, int, boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

heads() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the heads record component.

heads() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns the value of the heads record component.

height() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the height record component.

hidden() - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

Returns the value of the hidden record component.

hiddenProjection - Variable in class io.github.kirstenali.deepj.nn.RecurrentCell

hiddenSize() - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

hiddenSize() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns the value of the hiddenSize record component.

I

id - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

idToBytes() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the idToBytes record component.

idToSpecialToken() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

included(boolean[], int) - Static method in class io.github.kirstenali.deepj.tensor.CrossEntropyChecks

indexAdd(Tensor, int, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

indexAdd(Tensor, int, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

indexAdd(Tensor, int, int[], Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

indexAddInPlace(int, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

IndexedResponseTextDataset - Class in io.github.kirstenali.deepj.data

IndexedResponseTextDataset(List<ResponseOnlyTextDataset.Source>, Tokenizer, int, long) - Constructor for class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

indexSelect(int, int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

indexSelect(Tensor, int, int[]) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

indexSelect(Tensor, int, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

indexSelect(Tensor, int, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

indexSelect(Tensor, int, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

indexSelectGradient(Tensor, int[], int, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

indexSelectGradient(Tensor, int[], int, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

indexSelectGradient(Tensor, int[], int, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

InferenceContext - Class in io.github.kirstenali.deepj.inference

Cache slots follow logical attention visits, so each repeated Loop pass has its own state.

InferenceContext(int) - Constructor for class io.github.kirstenali.deepj.inference.InferenceContext

InferenceContext(int, boolean) - Constructor for class io.github.kirstenali.deepj.inference.InferenceContext

initial() - Static method in record class io.github.kirstenali.deepj.training.TrainingProgress

initialState(int, DType) - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

initScale() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the initScale record component.

initScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the initScale record component.

initScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the initScale record component.

initScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the initScale record component.

initScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the initScale record component.

initScale() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the initScale record component.

initScale() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

inner() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the inner record component.

inputProjection - Variable in class io.github.kirstenali.deepj.nn.RecurrentCell

inputSize() - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

inspect(Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

INSTRUCTION_RESPONSE - Enum constant in enum class io.github.kirstenali.deepj.nn.transformer.PromptFormat

io.github.kirstenali.deepj.activations - package io.github.kirstenali.deepj.activations

io.github.kirstenali.deepj.autograd - package io.github.kirstenali.deepj.autograd

io.github.kirstenali.deepj.chatui - package io.github.kirstenali.deepj.chatui

io.github.kirstenali.deepj.cli - package io.github.kirstenali.deepj.cli

io.github.kirstenali.deepj.concurrent - package io.github.kirstenali.deepj.concurrent

io.github.kirstenali.deepj.data - package io.github.kirstenali.deepj.data

io.github.kirstenali.deepj.examples - package io.github.kirstenali.deepj.examples

io.github.kirstenali.deepj.inference - package io.github.kirstenali.deepj.inference

io.github.kirstenali.deepj.layers - package io.github.kirstenali.deepj.layers

io.github.kirstenali.deepj.layers.transformer - package io.github.kirstenali.deepj.layers.transformer

io.github.kirstenali.deepj.layers.transformer.attention - package io.github.kirstenali.deepj.layers.transformer.attention

io.github.kirstenali.deepj.layers.transformer.blocks - package io.github.kirstenali.deepj.layers.transformer.blocks

io.github.kirstenali.deepj.layers.transformer.norm - package io.github.kirstenali.deepj.layers.transformer.norm

io.github.kirstenali.deepj.loss - package io.github.kirstenali.deepj.loss

io.github.kirstenali.deepj.nn - package io.github.kirstenali.deepj.nn

Neural networks composed from DeepJ tensors, layers, autograd and backend operations.

io.github.kirstenali.deepj.nn.transformer - package io.github.kirstenali.deepj.nn.transformer

io.github.kirstenali.deepj.nn.transformer.loop - package io.github.kirstenali.deepj.nn.transformer.loop

io.github.kirstenali.deepj.nn.transformer.nova - package io.github.kirstenali.deepj.nn.transformer.nova

io.github.kirstenali.deepj.nn.transformer.orbit - package io.github.kirstenali.deepj.nn.transformer.orbit

io.github.kirstenali.deepj.nn.transformer.origin - package io.github.kirstenali.deepj.nn.transformer.origin

io.github.kirstenali.deepj.nn.transformer.prism - package io.github.kirstenali.deepj.nn.transformer.prism

io.github.kirstenali.deepj.optimisers - package io.github.kirstenali.deepj.optimisers

io.github.kirstenali.deepj.persistence - package io.github.kirstenali.deepj.persistence

io.github.kirstenali.deepj.publishing - package io.github.kirstenali.deepj.publishing

io.github.kirstenali.deepj.tensor - package io.github.kirstenali.deepj.tensor

io.github.kirstenali.deepj.tensor.cpu - package io.github.kirstenali.deepj.tensor.cpu

io.github.kirstenali.deepj.tensor.metal - package io.github.kirstenali.deepj.tensor.metal

io.github.kirstenali.deepj.tokenizers - package io.github.kirstenali.deepj.tokenizers

io.github.kirstenali.deepj.tokenizers.bpe - package io.github.kirstenali.deepj.tokenizers.bpe

io.github.kirstenali.deepj.training - package io.github.kirstenali.deepj.training

io.github.kirstenali.deepj.transformer - package io.github.kirstenali.deepj.transformer

io.github.kirstenali.deepj.transformer.embeddings - package io.github.kirstenali.deepj.transformer.embeddings

isAvailable() - Static method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

isContiguous() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

isEmpty() - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

isEnabled() - Static method in class io.github.kirstenali.deepj.autograd.GradMode

isEndOfSequence(int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

isEndOfSequence(int) - Method in interface io.github.kirstenali.deepj.tokenizers.Tokenizer

isModelLoaded() - Method in interface io.github.kirstenali.deepj.chatui.ChatService

isModelLoaded() - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

isModelLoaded() - Method in class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

isModelLoaded() - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

isParallelEnabled() - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

isTraining() - Method in class io.github.kirstenali.deepj.nn.Module

item() - Method in class io.github.kirstenali.deepj.tensor.Tensor

K

key() - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Returns the value of the key record component.

kvHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the kvHeads record component.

kvHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the kvHeads record component.

kvRank() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the kvRank record component.

kvRank() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the kvRank record component.

kvRank(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

L

l2Norm(List<Tensor>) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

l2Norm(List<Tensor>) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

language() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the language record component.

lastLoss() - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Returns the value of the lastLoss record component.

lastLoss() - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Returns the value of the lastLoss record component.

Layer - Interface in io.github.kirstenali.deepj.layers

layerNorm(Tensor, Tensor, Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.NormalizationOps

LayerNorm1D - Class in io.github.kirstenali.deepj.layers.transformer.norm

LayerNorm1D(int) - Constructor for class io.github.kirstenali.deepj.layers.transformer.norm.LayerNorm1D

LayerNorm1D(int, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.norm.LayerNorm1D

layerNormBackward(Tensor, Tensor, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

layerNormBackward(Tensor, Tensor, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

layerNormBackward(Tensor, Tensor, Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

layers() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the layers record component.

layers() - Method in class io.github.kirstenali.deepj.nn.Sequential

layers() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns the value of the layers record component.

layerStatistics(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

layout(Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

layoutAdamW(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutBinary(Tensor, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutBinaryInto(Tensor, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutCausal(Tensor, int, float, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutCrossEntropy(Tensor, int[], boolean[], int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutIndexAdd(Tensor, int, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutIndexGradient(Tensor, int[], int, int[]) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutIndexSelect(Tensor, int, int[]) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutMatmul(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutMaximumGradient(Tensor, Tensor, Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutNormalizationBackward(Tensor, Tensor, Tensor, Tensor, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutNormalize(Tensor, Tensor, Tensor, float, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

LayoutOps - Class in io.github.kirstenali.deepj.autograd

layoutReduction(Tensor, int[], boolean, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutRotary(Tensor, Tensor, Tensor, int, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutSliceGradient(Tensor, int[], int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutSoftmax(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutSwiGluBackward(Tensor, Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutTotal(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutUnary(Tensor, int, float, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

layoutUnaryInto(Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

lazyZeros(DType, int...) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Zeros that stay unallocated until first use; a device zero-fill is queued only when a kernel consumes them.

learningRate() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the learningRate record component.

learningRate(int) - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

left() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Returns the value of the left record component.

leftRows() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the leftRows record component.

length() - Method in record class io.github.kirstenali.deepj.data.TextFileRange

length(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.SequenceLayout

license() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the license record component.

limitations() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the limitations record component.

linear(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.layers.ProjectionOps

Linear - Class in io.github.kirstenali.deepj.layers

Linear(int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.Linear

Linear(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.Linear

lmHead - Variable in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

load(DecoderOnlyModel, AdamW, StatefulBatchSource, CosineLearningRateSchedule, Path) - Static method in class io.github.kirstenali.deepj.persistence.TrainingCheckpoint

load(Path) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

load(Path) - Static method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

load(Path) - Static method in class io.github.kirstenali.deepj.nn.transformer.ModelLoader

load(Path) - Method in interface io.github.kirstenali.deepj.persistence.Persistable

load(Path) - Static method in class io.github.kirstenali.deepj.tokenizers.bpe.BPEModelIO

load(Path, Layer) - Static method in class io.github.kirstenali.deepj.nn.NetworkWeights

load(List<Parameter>, AdamW, StatefulBatchSource, CosineLearningRateSchedule, Path) - Static method in class io.github.kirstenali.deepj.persistence.TrainingCheckpoint

load(List<Parameter>, Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

load(List<Parameter>, Path, TransformerConfig) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

loadModel(Path) - Method in interface io.github.kirstenali.deepj.chatui.ChatService

loadModel(Path) - Method in class io.github.kirstenali.deepj.chatui.DeepJChatService

loadModel(Path) - Method in class io.github.kirstenali.deepj.chatui.DeepJOriginChatService

loadModel(Path) - Method in class io.github.kirstenali.deepj.chatui.DeepJPrismChatService

log() - Method in class io.github.kirstenali.deepj.tensor.Tensor

log(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

log(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

log(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

log(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

logEvery() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the logEvery record component.

logInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

logInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

logInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

logInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

LoopedTransformer - Class in io.github.kirstenali.deepj.transformer

Repeats shared grouped-query attention and SwiGLU blocks; each pass saves its own activations.

LoopedTransformer(DeepJLoopConfig, Random) - Constructor for class io.github.kirstenali.deepj.transformer.LoopedTransformer

LoopedTransformer(DeepJLoopConfig, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.transformer.LoopedTransformer

loops() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the loops record component.

loss() - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Returns the value of the loss record component.

loss() - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Returns the value of the loss record component.

loss(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

loss(Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

loss(Tensor, int[], boolean[]) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdCrossEntropy

loss(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

loss(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.loss.LossFunction

loss(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.loss.MSELoss

LossFunction - Interface in io.github.kirstenali.deepj.loss

lossMask() - Method in record class io.github.kirstenali.deepj.data.Batch

Returns the value of the lossMask record component.

lr() - Method in class io.github.kirstenali.deepj.optimisers.AdamW

LSTM - Class in io.github.kirstenali.deepj.nn

Batch-first LSTM layer; compose layers with Sequential for stacked recurrence.

LSTM(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.LSTM

LSTM(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.LSTM

LSTMCell - Class in io.github.kirstenali.deepj.nn

LSTM with input/forget/candidate/output gate order and explicit cell memory.

LSTMCell(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.LSTMCell

LSTMCell(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.LSTMCell

M

main(String[]) - Static method in class io.github.kirstenali.deepj.chatui.ChatApp

main(String[]) - Static method in class io.github.kirstenali.deepj.cli.DeepJCli

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.BenchmarkDeepJ100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.BenchmarkDeepJLoop100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.EvaluateDeepJ100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.EvaluateDeepJ90M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.EvaluateDeepJ90MPretraining

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.EvaluateDeepJLoop100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.EvaluateDeepJPrismKnowledge

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.ExportFrisky90M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.FineTuneDeepJ100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.FineTuneDeepJ90M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.FineTuneDeepJLoop100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.FineTuneDeepJPrismKnowledge

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.FormatAlpacaText

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.GenerateFactDataset

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.MidTrainDeepJ100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.MidTrainDeepJ90M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.MidTrainDeepJLoop100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainDeepJ100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainDeepJ90M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainDeepJLoop100M

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainDeepJPrismKnowledge

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainSmallDeepJOrbit

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainSmallDeepJOrigin

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainSmallDeepJOriginWithBPE

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainSmallDeepJPrism

markUpdated(Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

mask(int) - Method in record class io.github.kirstenali.deepj.data.Batch

mask(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdPosition

masterWeights() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the masterWeights record component.

matches(Path) - Static method in class io.github.kirstenali.deepj.persistence.TrainingCheckpoint

materialize() - Method in class io.github.kirstenali.deepj.tensor.Tensor

materialize() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

materialize(Tensor) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

materializeTensor(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

materializeTensor(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

matmul(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

matmul(TensorBackend, Tensor, Tensor, int, boolean, boolean) - Static method in class io.github.kirstenali.deepj.tensor.PackedLayouts

matmul(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BinaryOps

matmul(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

matmul(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

matmul(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

matrix(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

matrixCols() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

matrixRows() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Checked rank-two dimensions for matrix kernels; other ranks are never implicitly flattened.

MatrixShapeChecks - Class in io.github.kirstenali.deepj.tensor

Common validation for the existing rank-two CPU and native attention kernels.

max(int) - Static method in class io.github.kirstenali.deepj.nn.Pool1D

max(int) - Static method in class io.github.kirstenali.deepj.nn.Pool2D

MAX - Enum constant in enum class io.github.kirstenali.deepj.nn.Pool2D.Reduction

maxAlongRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

maxAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

maxAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

maxAlongRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

maxAxes(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

maximum(int, boolean) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

Tied maxima share their gradient equally.

maximum(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.SelectionOps

Tied maxima share the row gradient equally.

maximum(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.autograd.AxisReductionOps

maximum(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdStatistics

maximumAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

maximumAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

maximumAxes(Tensor, int[], boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

maximumAxesBackward(Tensor, Tensor, Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

maximumAxesBackward(Tensor, Tensor, Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

maximumAxesBackward(Tensor, Tensor, Tensor, int[], boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

maximumGradient(Tensor, Tensor, Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdStatistics

maxSeqLen() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the maxSeqLen record component.

maxSeqLen() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the maxSeqLen record component.

maxSeqLen() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the maxSeqLen record component.

maxSeqLen() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the maxSeqLen record component.

maxSeqLen() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the maxSeqLen record component.

maxSeqLen() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

maxSeqLen(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

maxSeqLen(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

mean(int, boolean) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

mean(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ReductionOps

mean(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.autograd.AxisReductionOps

mean(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdReduction

meanAlongRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

meanAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

meanAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

meanAlongRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

meanAxes(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

meanAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

meanAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

meanAxes(Tensor, int[], boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

meanLoss() - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

meanPool(Tensor, boolean[][]) - Method in class io.github.kirstenali.deepj.nn.TransformerEncoder

merge(TensorBackend, Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.PackedLayouts

merge(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadLayout

mergeHeads(Tensor, int) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

mergeHeads(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

mergeHeads(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

mergeHeads(Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

merges() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the merges record component.

mergeToNewId() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the mergeToNewId record component.

metadata() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

metadata() - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Returns the value of the metadata record component.

MetalBackend - Class in io.github.kirstenali.deepj.tensor.metal

MetalBackend() - Constructor for class io.github.kirstenali.deepj.tensor.metal.MetalBackend

MidTrainDeepJ100M - Class in io.github.kirstenali.deepj.examples

Continues Nova on Frisky's unchanged broad instruction mixture.

MidTrainDeepJ90M - Class in io.github.kirstenali.deepj.examples

MidTrainDeepJLoop100M - Class in io.github.kirstenali.deepj.examples

Runs the midtrain stage using the existing Frisky dataset.

minimumLearningRate() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the minimumLearningRate record component.

minimumLearningRate() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns the value of the minimumLearningRate record component.

model() - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

MODEL_CARD_FILE - Static variable in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

MODEL_FILE - Static variable in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

ModelCard - Record Class in io.github.kirstenali.deepj.publishing

ModelCard(String, String, String, String, String, String) - Constructor for record class io.github.kirstenali.deepj.publishing.ModelCard

Creates an instance of a ModelCard record class.

modelConfig(int) - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

modelFormatVersion() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the modelFormatVersion record component.

ModelInspection - Record Class in io.github.kirstenali.deepj.persistence

Embedded settings and storage counts, read without allocating model weights.

ModelInspection(ModelSettings, long, int, long) - Constructor for record class io.github.kirstenali.deepj.persistence.ModelInspection

Creates an instance of a ModelInspection record class.

ModelLayout - Record Class in io.github.kirstenali.deepj.persistence

Validated model construction metadata,without allocating or converting tensor values.

ModelLayout(ModelSettings, Map<String, TensorDescriptor>) - Constructor for record class io.github.kirstenali.deepj.persistence.ModelLayout

Creates an instance of a ModelLayout record class.

ModelLoader - Class in io.github.kirstenali.deepj.nn.transformer

ModelSerializer - Class in io.github.kirstenali.deepj.persistence

ModelSettings - Record Class in io.github.kirstenali.deepj.persistence

ModelSettings(TransformerConfig, PromptFormat) - Constructor for record class io.github.kirstenali.deepj.persistence.ModelSettings

ModelSettings(TransformerConfig, PromptFormat, Map<String, String>) - Constructor for record class io.github.kirstenali.deepj.persistence.ModelSettings

Creates an instance of a ModelSettings record class.

Module - Class in io.github.kirstenali.deepj.nn

Additive layer base with recursive training mode and identity-deduplicated parameters.

Module() - Constructor for class io.github.kirstenali.deepj.nn.Module

MSELoss - Class in io.github.kirstenali.deepj.loss

Mean squared error over equal, nonempty shapes, with FLOAT32 loss and gradients.

MSELoss() - Constructor for class io.github.kirstenali.deepj.loss.MSELoss

MultiHeadLatentAttention - Class in io.github.kirstenali.deepj.layers.transformer.attention

MultiHeadLatentAttention(int, int, int, int, RotaryEmbedding, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

MultiHeadLatentAttention(int, int, int, int, RotaryEmbedding, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

MultiHeadSelfAttention - Class in io.github.kirstenali.deepj.layers.transformer.attention

MultiHeadSelfAttention(int, int, boolean, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

MultiHeadSelfAttention(int, int, boolean, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

multiply(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiply(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BinaryOps

multiply(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiply(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiply(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

multiply(Tensor, Tensor, boolean) - Static method in class io.github.kirstenali.deepj.autograd.BroadcastOps

multiplyBroadcastCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiplyBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiplyBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiplyBroadcastCols(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

multiplyBroadcastRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiplyBroadcastRows(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiplyBroadcastRows(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiplyBroadcastRows(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

multiplyInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiplyInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiplyInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiplyInPlace(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

multiplyScalar(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiplyScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiplyScalar(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiplyScalar(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

multiplyScalarInPlace(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

multiplyScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

multiplyScalarInPlace(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

multiplyScalarInPlace(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

N

name() - Method in class io.github.kirstenali.deepj.optimisers.Parameter

name() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the name record component.

NdSupervisedTraining - Class in io.github.kirstenali.deepj.nn

First-axis minibatching for images, sequences and matrices, using DeepJ's existing Trainer.

needsUpload - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

neg() - Method in class io.github.kirstenali.deepj.tensor.Tensor

neg(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

neg(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

neg(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

neg(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

negInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

negInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

negInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

negInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

NetworkWeights - Class in io.github.kirstenali.deepj.nn

Versioned parameter/buffer archives for a separately constructed network.

newAlias() - Method in class io.github.kirstenali.deepj.autograd.GradientState

Independent autograd metadata sharing only the physical storage's mutation version.

newChat() - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

newOutputBuffer(int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

newOutputBuffer(int, int, DType) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

newZeroBuffer(int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Queues explicit zero initialization before any operations consume this buffer.

newZeroBuffer(int, int, DType) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

next(Tensor, RecurrentState) - Method in class io.github.kirstenali.deepj.nn.GRUCell

next(Tensor, RecurrentState) - Method in class io.github.kirstenali.deepj.nn.LSTMCell

next(Tensor, RecurrentState) - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

next(Tensor, RecurrentState) - Method in class io.github.kirstenali.deepj.nn.RNNCell

nextBatch(int) - Method in interface io.github.kirstenali.deepj.data.BatchSource

nextBatch(int) - Method in class io.github.kirstenali.deepj.data.RandomAccessTextDataset

nextBatch(int) - Method in class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

nextBatch(int) - Method in class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset

nextBatch(int) - Method in class io.github.kirstenali.deepj.data.SequentialTextDataset

nextBatch(int) - Method in class io.github.kirstenali.deepj.data.TextDataset

nHeads - Variable in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

nHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the nHeads record component.

nHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the nHeads record component.

nHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the nHeads record component.

nHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the nHeads record component.

nHeads() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the nHeads record component.

nHeads() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

nHeads(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

nHeads(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

nHeads(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

nLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

nLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the nLayers record component.

nLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the nLayers record component.

nLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the nLayers record component.

nLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the nLayers record component.

nLayers() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

nLayers(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

nLayers(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

nLayers(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

node() - Method in class io.github.kirstenali.deepj.autograd.GradientState

noGrad() - Static method in class io.github.kirstenali.deepj.autograd.GradMode

normalization() - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Returns the value of the normalization record component.

normalizationBackward(Tensor, Tensor, Tensor, Tensor, boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

normalizationBackward(Tensor, Tensor, Tensor, Tensor, boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

normalizationBackward(Tensor, Tensor, Tensor, Tensor, boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

NormalizationChecks - Class in io.github.kirstenali.deepj.tensor

Final-axis normalization accepts vector gains and the historical 1-by-width parameter layout.

NormalizationOps - Class in io.github.kirstenali.deepj.autograd

NormalizationResult - Record Class in io.github.kirstenali.deepj.tensor

The output uses promoted storage; normalized values and denominators retain FLOAT32 accumulation.

NormalizationResult(Tensor, Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.NormalizationResult

Creates an instance of a NormalizationResult record class.

normalize(Tensor, Tensor, Tensor, float, boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

normalize(Tensor, Tensor, Tensor, float, boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

normalize(Tensor, Tensor, Tensor, float, boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

normalized() - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Returns the value of the normalized record component.

normalized() - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Returns the value of the normalized record component.

normF - Variable in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

NormLayer - Interface in io.github.kirstenali.deepj.layers.transformer.norm

numel() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

O

of(Tensor, Tensor, int, boolean, boolean) - Static method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

ones(int, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

ones(int, int) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

ones(DType, int...) - Static method in class io.github.kirstenali.deepj.tensor.TensorManipulations

ones(String, int, int) - Method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

onStep(int, float, float) - Method in interface io.github.kirstenali.deepj.training.Trainer.StepHook

OP_ADAMW_UPDATE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_ADD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_ADD_BROADCAST_COLS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_ADD_ROW_VECTOR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_ADD_SCALAR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_BATCHED_MATMUL - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CAUSAL_MASK - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CAUSAL_SOFTMAX - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CLAMP - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CROSS_ENTROPY_FUSED - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CROSS_ENTROPY_GRADIENT - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_CROSS_ENTROPY_LOSS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_DIVIDE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_DIVIDE_BROADCAST_COLS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_DIVIDE_SCALAR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_EXP - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_GATHER_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_GELU - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_GELU_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_HEAD_CACHE_WRITE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_HEAD_PREFIX_MATMUL - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYERNORM_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_ADAMW - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_ASSIGN - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_BINARY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_CAUSAL - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_CROSS_ENTROPY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_INDEX_ADD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_INDEX_SELECT - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_MATMUL - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_MAXIMUM_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_NORMALIZATION_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_NORMALIZE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_REDUCE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_ROTARY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_SOFTMAX - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_SWIGLU_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_TRANSFER - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LAYOUT_UNARY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_LOG - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MATMUL - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MAX_ALONG_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MEAN_ALONG_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MERGE_HEADS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MULTIPLY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MULTIPLY_BROADCAST_COLS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MULTIPLY_BROADCAST_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_MULTIPLY_SCALAR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_NEG - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_POW - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_RELU - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_RELU_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_RMS_NORM - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_RMS_NORM_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_ROTARY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SCALED_SOFTMAX_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SCATTER_ADD_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SCATTER_ADD_ROWS_ATOMIC - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SIGMOID - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SOFTMAX_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SOFTMAX_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SPLIT_HEADS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SQRT - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_STORAGE_COPY - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUBTRACT - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUBTRACT_BROADCAST_COLS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_ABS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_ALONG_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_SCALAR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_SQUARES - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SUM_SQUARES_SCALAR - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SWIGLU - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_SWIGLU_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_TANH - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_TILED_CAUSAL_ATTENTION - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_TILED_CAUSAL_ATTENTION_BACKWARD - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_TRANSPOSE - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

OP_VARIANCE_ALONG_ROWS - Static variable in class io.github.kirstenali.deepj.tensor.ComputeGraph

opcodeNames() - Static method in class io.github.kirstenali.deepj.tensor.GraphOpHistogram

open() - Static method in class io.github.kirstenali.deepj.tensor.TensorScope

openGeneration() - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

openGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

openGeneration(ContextWindowMode) - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

openGeneration(ContextWindowMode) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

OptimizerChecks - Class in io.github.kirstenali.deepj.tensor

Adam updates accumulate in independent FLOAT32 parameter/master and moment buffers.

output() - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Returns the value of the output record component.

output() - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Returns the value of the output record component.

output() - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Returns the value of the output record component.

outputDirectory() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Returns the value of the outputDirectory record component.

outputHeight(int) - Method in record class io.github.kirstenali.deepj.nn.Window2D

outputType(Tensor, Tensor, Tensor, boolean) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

outputWidth(int) - Method in record class io.github.kirstenali.deepj.nn.Window2D

P

pack(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.HeadLayout

PackedLayouts - Class in io.github.kirstenali.deepj.tensor

Explicit compatibility for packed matrix APIs,delegating their work to the same ND operators.

PackedMatmulShape - Record Class in io.github.kirstenali.deepj.tensor

Logical dimensions for the compatibility API; never inferred from a GPU allocation's physical rows.

PackedMatmulShape(int, int, int, int, int, boolean, boolean) - Constructor for record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Creates an instance of a PackedMatmulShape record class.

packF32(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorAdapters

paddingHeight() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the paddingHeight record component.

paddingWidth() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the paddingWidth record component.

parameter(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

Parameter - Class in io.github.kirstenali.deepj.optimisers

Parameter(Tensor) - Constructor for class io.github.kirstenali.deepj.optimisers.Parameter

Parameter(String, Tensor) - Constructor for class io.github.kirstenali.deepj.optimisers.Parameter

parameterFactory() - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

ParameterFactory - Class in io.github.kirstenali.deepj.optimisers

Explicit parameter storage policy: a requested dtype applies to multi-row parameters; one-row parameters remain FLOAT32.

ParameterFactory(DType) - Constructor for class io.github.kirstenali.deepj.optimisers.ParameterFactory

ParameterFactory(Map<String, TensorDescriptor>) - Constructor for class io.github.kirstenali.deepj.optimisers.ParameterFactory

ParameterOptimizer - Interface in io.github.kirstenali.deepj.optimisers

parameters() - Method in class io.github.kirstenali.deepj.layers.BiasFreeLinear

parameters() - Method in class io.github.kirstenali.deepj.layers.Linear

parameters() - Method in class io.github.kirstenali.deepj.layers.TiedEmbeddingProjection

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.norm.LayerNorm1D

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.norm.RMSNorm1D

parameters() - Method in class io.github.kirstenali.deepj.layers.transformer.SwiGLULayer

parameters() - Method in class io.github.kirstenali.deepj.nn.BatchNorm

parameters() - Method in class io.github.kirstenali.deepj.nn.Conv2D

parameters() - Method in class io.github.kirstenali.deepj.nn.FNN

parameters() - Method in class io.github.kirstenali.deepj.nn.Module

parameters() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

parameters() - Method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

parameters() - Method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

parameters() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns the value of the parameters record component.

parameters() - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

Returns the value of the parameters record component.

parameters() - Method in interface io.github.kirstenali.deepj.persistence.Persistable

parameters() - Method in interface io.github.kirstenali.deepj.training.Trainable

parameters() - Method in class io.github.kirstenali.deepj.transformer.embeddings.Embedding

parameters() - Method in class io.github.kirstenali.deepj.transformer.embeddings.PositionalEmbedding

parameters() - Method in class io.github.kirstenali.deepj.transformer.LoopedTransformer

parameters() - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

parameters(ParameterFactory) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

parameters(ParameterFactory) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

parameters(ParameterFactory) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

path() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Returns the value of the path record component.

peakLearningRate() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the peakLearningRate record component.

peakLearningRate() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns the value of the peakLearningRate record component.

permute(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

permute(Tensor, int...) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

permute(Tensor, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

perplexity() - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Returns the value of the perplexity record component.

Persistable - Interface in io.github.kirstenali.deepj.persistence

PLAIN - Enum constant in enum class io.github.kirstenali.deepj.nn.transformer.PromptFormat

Pool1D - Class in io.github.kirstenali.deepj.nn

Max/average pooling of [batch, channels, length] inputs.

Pool1D(int, int, int, Pool2D.Reduction) - Constructor for class io.github.kirstenali.deepj.nn.Pool1D

Pool2D - Class in io.github.kirstenali.deepj.nn

Floor-mode max/average pooling of channel-first images.

Pool2D(Window2D, Pool2D.Reduction) - Constructor for class io.github.kirstenali.deepj.nn.Pool2D

Pool2D.Reduction - Enum Class in io.github.kirstenali.deepj.nn

position() - Method in class io.github.kirstenali.deepj.inference.InferenceContext

PositionalEmbedding - Class in io.github.kirstenali.deepj.transformer.embeddings

PositionalEmbedding(int, int, Random) - Constructor for class io.github.kirstenali.deepj.transformer.embeddings.PositionalEmbedding

PositionalEmbedding(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.transformer.embeddings.PositionalEmbedding

positions(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CrossEntropyChecks

pow(float) - Method in class io.github.kirstenali.deepj.tensor.Tensor

pow(Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

pow(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

pow(Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

pow(Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

prefix(TensorBackend, Tensor, int, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadCacheOps

preludeLayers() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the preludeLayers record component.

PreparationResult(FormatAlpacaText.FormatResult, GenerateFactDataset.GenerationResult, PrepareKnowledgeCorpus.SplitResult, PrepareKnowledgeCorpus.SplitResult, long) - Constructor for record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Creates an instance of a PreparationResult record class.

prepare(Path, Path, Path, int, int, int) - Static method in class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus

PrepareKnowledgeCorpus - Class in io.github.kirstenali.deepj.examples

PrepareKnowledgeCorpus.PreparationResult - Record Class in io.github.kirstenali.deepj.examples

PrepareKnowledgeCorpus.SplitResult - Record Class in io.github.kirstenali.deepj.examples

print(Tensor, String) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

print(String) - Method in class io.github.kirstenali.deepj.tensor.Tensor

Projection - Interface in io.github.kirstenali.deepj.layers

ProjectionOps - Class in io.github.kirstenali.deepj.layers

Independent leading positions share one projection; public activation axes remain intact.

promote(DType, DType) - Static method in enum class io.github.kirstenali.deepj.tensor.DType

Different formats promote to FLOAT32; identical formats retain their dtype.

prompt() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns the value of the prompt record component.

promptFormat() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

promptFormat() - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Returns the value of the promptFormat record component.

PromptFormat - Enum Class in io.github.kirstenali.deepj.nn.transformer

Q

qkNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the qkNorm record component.

qkNorm() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the qkNorm record component.

qRank() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the qRank record component.

qRank() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the qRank record component.

qRank(int) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

query() - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Returns the value of the query record component.

R

random(int, int, Random) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

random(int, int, Random) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

random(DType, Random, int...) - Static method in class io.github.kirstenali.deepj.tensor.TensorManipulations

random(String, int, int, Random) - Method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

random(Random) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

random(Random) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

random(Random) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

RandomAccessTextDataset - Class in io.github.kirstenali.deepj.data

RandomAccessTextDataset(Path, Tokenizer, int, long) - Constructor for class io.github.kirstenali.deepj.data.RandomAccessTextDataset

RandomAccessTextDataset(Path, Tokenizer, int, long, TextFileRange) - Constructor for class io.github.kirstenali.deepj.data.RandomAccessTextDataset

randomState() - Method in class io.github.kirstenali.deepj.data.RandomAccessTextDataset

randomState() - Method in class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

randomState() - Method in class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset

randomState() - Method in class io.github.kirstenali.deepj.data.SequentialTextDataset

randomState() - Method in interface io.github.kirstenali.deepj.data.StatefulBatchSource

rank() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

rawData() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Unsafe backing access bypasses mutation tracking; use managed Tensor get/set for application code.

read(DataInput) - Static method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

readInto(DataInput, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

readValues(DataInput, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

readValuesToBackend(DataInput, DType, int...) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

Explicit checkpoint transfer; private host staging is released after an independent backend copy.

REBUILD - Enum constant in enum class io.github.kirstenali.deepj.nn.transformer.ContextWindowMode

record(Tensor, BackwardRule, Tensor...) - Static method in class io.github.kirstenali.deepj.autograd.Autograd

recordAdamWUpdate(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, float, float, float, float, float, float, float, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordBatchedMatmul(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int, int, int, boolean, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordBinary(int, GpuBuffer, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalAttention(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalAttention(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalAttentionBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalAttentionBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalMask(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCausalSoftmax(GpuBuffer, GpuBuffer, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordClamp(GpuBuffer, GpuBuffer, float, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordColBroadcast(int, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCount() - Method in class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

recordCrossEntropy(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCrossEntropyGradient(GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordCrossEntropyLoss(GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordGatherRows(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordHeadCacheWrite(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordHeadCacheWrite(GpuBuffer, GpuBuffer, int, int, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordHeadPermutation(int, GpuBuffer, GpuBuffer, int, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordHeadPrefixMatmul(GpuBuffer, GpuBuffer, GpuBuffer, int, int, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordHeadPrefixMatmul(GpuBuffer, GpuBuffer, GpuBuffer, int, int, boolean, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordLayerNormBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordMatmul(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordMultiplyScalar(GpuBuffer, GpuBuffer, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordPow(GpuBuffer, GpuBuffer, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordReduction(int, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordRmsNorm(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordRmsNormBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordRotary(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int, int, boolean) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordRowBroadcast(int, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSaved(Tensor, BackwardRule, Tensor[], Tensor...) - Static method in class io.github.kirstenali.deepj.autograd.Autograd

recordScalarUnary(int, GpuBuffer, GpuBuffer, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordScaledSoftmaxBackward(GpuBuffer, GpuBuffer, GpuBuffer, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordScatterAddRows(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordScatterAddRowsAtomic(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSoftmaxBackward(GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSoftmaxRows(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordStorageCopy(GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSumAbs(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSumScalar(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSumSquares(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSumSquaresScalar(GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSwiGlu(GpuBuffer, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordSwiGluBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordTranspose(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

recordUnary(int, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Recurrent - Class in io.github.kirstenali.deepj.nn

Batch-first recurrent sequence execution, with optional lengths and caller-owned initial/final state.

Recurrent(RecurrentCell) - Constructor for class io.github.kirstenali.deepj.nn.Recurrent

RecurrentCell - Class in io.github.kirstenali.deepj.nn

Shared input/state validation and DeepJ linear projections for recurrent cells.

RecurrentCell(int, int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.RecurrentCell

RecurrentResult - Record Class in io.github.kirstenali.deepj.nn

Batch-first sequence output and the final state of a recurrent layer.

RecurrentResult(Tensor, RecurrentState) - Constructor for record class io.github.kirstenali.deepj.nn.RecurrentResult

Creates an instance of a RecurrentResult record class.

RecurrentState - Record Class in io.github.kirstenali.deepj.nn

Explicit caller-owned state.

RecurrentState(Tensor) - Constructor for record class io.github.kirstenali.deepj.nn.RecurrentState

RecurrentState(Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.nn.RecurrentState

Creates an instance of a RecurrentState record class.

ReductionOps - Class in io.github.kirstenali.deepj.autograd

reference() - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Returns the value of the reference record component.

releaseAll() - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

releaseBuffers(int[], int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

releaseDeviceBufferRetention() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

releaseEvery() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the releaseEvery record component.

releaseResources() - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

releaseResources() - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

releaseTemporary() - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

releaseTemporaryResources() - Method in class io.github.kirstenali.deepj.tensor.BackwardBuffers

Releases this private computation's scratch without invalidating unrelated caller tensors.

releaseTemporaryResources() - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

releaseTemporaryResources() - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

relu() - Static method in class io.github.kirstenali.deepj.nn.Activation

relu(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

relu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

relu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

relu(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

ReLU - Class in io.github.kirstenali.deepj.activations

ReLU() - Constructor for class io.github.kirstenali.deepj.activations.ReLU

reluActivation() - Method in class io.github.kirstenali.deepj.tensor.Tensor

reluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

reluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

reluBackward(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

reluInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

reluInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

reluInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

reluInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

require(boolean) - Method in class io.github.kirstenali.deepj.autograd.GradientState

require(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadLayout

requireCausal(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.MatrixShapeChecks

required() - Method in class io.github.kirstenali.deepj.autograd.GradientState

requireMatches(DType, int[]) - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

requireMatches(Tensor) - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

requireMatches(List<Parameter>) - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

requireMatrix(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.MatrixShapeChecks

requirePrefix(Tensor, int, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadCacheOps

requireRotary(Tensor, Tensor, Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.MatrixShapeChecks

requireRowMaximum(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.MatrixShapeChecks

requireSameShape(Tensor, Tensor, String) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

requireScale(float) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdPosition

requiresGrad() - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

requiresGrad(boolean) - Method in class io.github.kirstenali.deepj.tensor.Tensor

requireState(Tensor, Tensor, Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.OptimizerChecks

requireTargets(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.tensor.CrossEntropyChecks

requireTargetsMatchRows(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

requireValid(int[], int[], DType[], int) - Static method in class io.github.kirstenali.deepj.tensor.GpuAllocations

requireWidth(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.SequenceLayout

requireWrite(Tensor, Tensor, int, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadCacheOps

reset() - Static method in class io.github.kirstenali.deepj.tensor.GraphOpHistogram

reshape(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

reshape(Tensor, int...) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

reshape(Tensor, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Reshape - Class in io.github.kirstenali.deepj.nn

Reshapes the full tensor, with at most one inferred (-1) dimension.

Reshape(int...) - Constructor for class io.github.kirstenali.deepj.nn.Reshape

response() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns the value of the response record component.

ResponseOnlyTextDataset - Class in io.github.kirstenali.deepj.data

ResponseOnlyTextDataset(List<ResponseOnlyTextDataset.Source>, Tokenizer, int, long) - Constructor for class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset

ResponseOnlyTextDataset.Example - Record Class in io.github.kirstenali.deepj.data

ResponseOnlyTextDataset.Source - Record Class in io.github.kirstenali.deepj.data

responseStart() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns the value of the responseStart record component.

restore(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

Explicit host-side restoration after file validation; not an operator fallback.

restoreRandomState(long) - Method in class io.github.kirstenali.deepj.data.RandomAccessTextDataset

restoreRandomState(long) - Method in class io.github.kirstenali.deepj.data.IndexedResponseTextDataset

restoreRandomState(long) - Method in class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset

restoreRandomState(long) - Method in class io.github.kirstenali.deepj.data.SequentialTextDataset

restoreRandomState(long) - Method in interface io.github.kirstenali.deepj.data.StatefulBatchSource

restoreState(List<Parameter>, AdamW.State) - Method in class io.github.kirstenali.deepj.optimisers.AdamW

result(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

resumeCheckpoint() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Returns the value of the resumeCheckpoint record component.

retainDeviceBuffer() - Method in class io.github.kirstenali.deepj.tensor.Tensor

retainsDeviceBuffer() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

right() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Returns the value of the right record component.

rightRows() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the rightRows record component.

rms() - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Returns the value of the rms record component.

rmsNorm(Tensor, Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.NormalizationOps

rmsNorm(Tensor, Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

rmsNorm(Tensor, Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

rmsNorm(Tensor, Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

RMSNorm1D - Class in io.github.kirstenali.deepj.layers.transformer.norm

RMSNorm1D(int) - Constructor for class io.github.kirstenali.deepj.layers.transformer.norm.RMSNorm1D

RMSNorm1D(int, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.norm.RMSNorm1D

rmsNormBackward(Tensor, Tensor, Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

rmsNormBackward(Tensor, Tensor, Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

rmsNormBackward(Tensor, Tensor, Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

RmsNormResult - Record Class in io.github.kirstenali.deepj.tensor

RmsNormResult(Tensor, Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.RmsNormResult

Creates an instance of a RmsNormResult record class.

RNN - Class in io.github.kirstenali.deepj.nn

Batch-first RNN layer; compose layers with Sequential for stacked recurrence.

RNN(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.RNN

RNN(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.RNN

RNNCell - Class in io.github.kirstenali.deepj.nn

Elman recurrent cell with tanh activation.

RNNCell(int, int, Random) - Constructor for class io.github.kirstenali.deepj.nn.RNNCell

RNNCell(int, int, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.RNNCell

ROLLING - Enum constant in enum class io.github.kirstenali.deepj.nn.transformer.ContextWindowMode

RoPEMultiHeadSelfAttention - Class in io.github.kirstenali.deepj.layers.transformer.attention

RoPEMultiHeadSelfAttention(int, int, boolean, RotaryEmbedding, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.RoPEMultiHeadSelfAttention

RoPEMultiHeadSelfAttention(int, int, boolean, RotaryEmbedding, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.attention.RoPEMultiHeadSelfAttention

rotary(Tensor, Tensor, Tensor, int, boolean) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

rotary(Tensor, Tensor, Tensor, int, boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdPosition

rotary(Tensor, Tensor, Tensor, int, boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

rotary(Tensor, Tensor, Tensor, int, boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

RotaryEmbedding - Class in io.github.kirstenali.deepj.transformer.embeddings

RotaryEmbedding(int, int) - Constructor for class io.github.kirstenali.deepj.transformer.embeddings.RotaryEmbedding

rotate(RotaryEmbedding, Tensor, int, int) - Method in class io.github.kirstenali.deepj.inference.InferenceContext

rowData(int) - Method in class io.github.kirstenali.deepj.tensor.Tensor

rows - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

rows() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the rows record component.

rows(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ReductionOps

run() - Method in class io.github.kirstenali.deepj.cli.DeepJCli

run(Tensor, RecurrentState, int[]) - Method in class io.github.kirstenali.deepj.nn.Recurrent

runningMean() - Method in class io.github.kirstenali.deepj.nn.BatchNorm

runningVariance() - Method in class io.github.kirstenali.deepj.nn.BatchNorm

S

sameMatrices(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

sample(Tensor, int, Random) - Static method in class io.github.kirstenali.deepj.tensor.CpuRowAccess

sampleMiB() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Returns the value of the sampleMiB record component.

sampleRows(Tensor, int, Random) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sampleRows(Tensor, int, Random) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

save(DecoderOnlyModel, AdamW, StatefulBatchSource, TrainingProgress, CosineLearningRateSchedule, Path) - Static method in class io.github.kirstenali.deepj.persistence.TrainingCheckpoint

save(Path) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

save(Path) - Method in interface io.github.kirstenali.deepj.persistence.Persistable

save(Path, Layer) - Static method in class io.github.kirstenali.deepj.nn.NetworkWeights

save(Path, BPEModel) - Static method in class io.github.kirstenali.deepj.tokenizers.bpe.BPEModelIO

save(List<Parameter>, AdamW, StatefulBatchSource, TrainingProgress, CosineLearningRateSchedule, Path) - Static method in class io.github.kirstenali.deepj.persistence.TrainingCheckpoint

save(List<Parameter>, Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

save(List<Parameter>, Path, ModelSettings) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

savedForBackward() - Method in class io.github.kirstenali.deepj.autograd.GradientState

scale(Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

scatterAddRows(Tensor, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

scatterAddRows(Tensor, int[], Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

scatterAddRows(Tensor, int[], Tensor) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

scatterAddRows(Tensor, int[], Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

scope(String) - Method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

secondMoments() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the secondMoments record component.

seed() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns the value of the seed record component.

seed() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns the value of the seed record component.

seed(long) - Method in class io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

seed(long) - Method in class io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

seed(long) - Method in class io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

select(int, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

select(Tensor, int[], int) - Static method in class io.github.kirstenali.deepj.tensor.CpuRowAccess

select(Tensor, int, int) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

select(Tensor, int, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

select(Tensor, int, int[]) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdIndexing

SelectionOps - Class in io.github.kirstenali.deepj.autograd

send(String, int, double, int, long) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

send(String, int, double, int, long, double) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

sendWithSettings(String, int, double, int, long, double) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

sendWithWindowSettings(String, int, double, int, long, double, boolean) - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

seqLen() - Method in class io.github.kirstenali.deepj.data.TextDataset

sequence() - Method in record class io.github.kirstenali.deepj.nn.RecurrentResult

Returns the value of the sequence record component.

SequenceLayout - Class in io.github.kirstenali.deepj.tensor

Model activations use[batch,time,width]; historical single-sequence matrices remain explicit.

sequenceLength() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns the value of the sequenceLength record component.

sequenceLength() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns the value of the sequenceLength record component.

Sequential - Class in io.github.kirstenali.deepj.nn

Ordered composition of existing or new DeepJ layers.

Sequential(Layer...) - Constructor for class io.github.kirstenali.deepj.nn.Sequential

SequentialTextDataset - Class in io.github.kirstenali.deepj.data

SequentialTextDataset(Path, Tokenizer, int) - Constructor for class io.github.kirstenali.deepj.data.SequentialTextDataset

SequentialTextDataset(Path, Tokenizer, int, TextFileRange) - Constructor for class io.github.kirstenali.deepj.data.SequentialTextDataset

set(int, int, float) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

set(Tensor, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

setAt(float, int...) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

setBackend(TensorBackend) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

setGpuTag(Object) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

setGradient(Tensor) - Method in class io.github.kirstenali.deepj.optimisers.Parameter

setLr(float) - Method in class io.github.kirstenali.deepj.optimisers.AdamW

setMetadata(Map<String, String>) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

setNumThreads(int) - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

setParallelEnabled(boolean) - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

setPromptFormat(PromptFormat) - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

setRow(int, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.Tensor

setRow(Tensor, int, Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

settings() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns the value of the settings record component.

settings() - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

Returns the value of the settings record component.

settings(Path) - Static method in class io.github.kirstenali.deepj.persistence.ModelSerializer

Settings(int, int, int, long) - Constructor for record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Creates an instance of a Settings record class.

setTraining(boolean) - Method in class io.github.kirstenali.deepj.nn.Module

shape() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Returns the value of the shape record component.

shape() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

shift(Tensor, float) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

shutdown() - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

sigmoid() - Static method in class io.github.kirstenali.deepj.nn.Activation

sigmoid(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

sigmoid(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sigmoid(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sigmoid(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Sigmoid - Class in io.github.kirstenali.deepj.activations

Sigmoid() - Constructor for class io.github.kirstenali.deepj.activations.Sigmoid

sigmoidActivation() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sigmoidInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sigmoidInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sigmoidInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sigmoidInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

SiLU - Class in io.github.kirstenali.deepj.activations

SiLU() - Constructor for class io.github.kirstenali.deepj.activations.SiLU

size() - Method in class io.github.kirstenali.deepj.data.TextDataset

size(int) - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

skipped() - Method in record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Returns the value of the skipped record component.

slice(int, int, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

slice(int, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

slice(Tensor, int, int, int, int) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

slice(Tensor, int, int, int, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sliceGradient(Tensor, int[], int, int, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sliceGradient(Tensor, int[], int, int, int) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdReduction

sliceGradient(Tensor, int[], int, int, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sliceGradient(Tensor, int[], int, int, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sliceRows(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sliceRows(Tensor, int[]) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sliceRows(Tensor, int[]) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sliceRows(Tensor, int[], int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sliceRows(Tensor, int[], int) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

snapshot() - Static method in class io.github.kirstenali.deepj.tensor.GraphOpHistogram

softmax(int) - Static method in class io.github.kirstenali.deepj.nn.Activation

softmax(int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

softmax(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

softmax(Tensor, int) - Static method in class io.github.kirstenali.deepj.autograd.AxisReductionOps

softmax(Tensor, int, float) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdPosition

Softmax - Class in io.github.kirstenali.deepj.activations

Softmax() - Constructor for class io.github.kirstenali.deepj.activations.Softmax

softmaxAxis(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

softmaxAxis(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

softmaxAxis(Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

softmaxAxisBackward(Tensor, Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

softmaxBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

softmaxBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

softmaxBackward(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

softmaxBackward(Tensor, Tensor, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

softmaxBackward(Tensor, Tensor, float) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

softmaxRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

softmaxRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

softmaxRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

softmaxRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Source(Path, int) - Constructor for record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Creates an instance of a Source record class.

specialTokenToId() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the specialTokenToId record component.

split(TensorBackend, Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.PackedLayouts

split(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadLayout

splitHeads(Tensor, int) - Static method in class io.github.kirstenali.deepj.autograd.LayoutOps

splitHeads(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

splitHeads(Tensor, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

splitHeads(Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

SplitResult(long, long) - Constructor for record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Creates an instance of a SplitResult record class.

sqrt() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sqrt(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

sqrt(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sqrt(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sqrt(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sqrtInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sqrtInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sqrtInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sqrtInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

stack - Variable in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

stack(int, Tensor...) - Static method in class io.github.kirstenali.deepj.nn.TensorJoin

start(Stage) - Method in class io.github.kirstenali.deepj.chatui.BaseChatApp

startInclusive() - Method in record class io.github.kirstenali.deepj.data.TextFileRange

Returns the value of the startInclusive record component.

state() - Method in record class io.github.kirstenali.deepj.nn.RecurrentResult

Returns the value of the state record component.

state(List<Parameter>) - Method in class io.github.kirstenali.deepj.optimisers.AdamW

State(long, float, float, float, float, float, List<Tensor>, List<Tensor>) - Constructor for record class io.github.kirstenali.deepj.optimisers.AdamW.State

State(long, float, float, float, float, float, List<Tensor>, List<Tensor>, Map<Integer, Tensor>) - Constructor for record class io.github.kirstenali.deepj.optimisers.AdamW.State

Creates an instance of a State record class.

StatefulBatchSource - Interface in io.github.kirstenali.deepj.data

StatefulTrainingDataset - Interface in io.github.kirstenali.deepj.data

statistics(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

statisticsShape(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

step() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the step record component.

step(Tensor, RecurrentState) - Method in class io.github.kirstenali.deepj.nn.RecurrentCell

step(List<Parameter>) - Method in class io.github.kirstenali.deepj.optimisers.AdamW

step(List<Parameter>) - Method in interface io.github.kirstenali.deepj.optimisers.ParameterOptimizer

stepCount() - Method in class io.github.kirstenali.deepj.optimisers.AdamW

steps() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the steps record component.

steps() - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Returns the value of the steps record component.

stop() - Method in class io.github.kirstenali.deepj.chatui.ChatWindow

storageBytes() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Physical bytes in the shared allocation; views may address only part of it.

storageBytes(int) - Method in enum class io.github.kirstenali.deepj.tensor.DType

storageCode() - Method in enum class io.github.kirstenali.deepj.tensor.DType

Stable storage identifier shared by checkpoint and native buffer descriptors.

storageOffset() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

storageSavedForBackward() - Method in class io.github.kirstenali.deepj.autograd.GradientState

stream(CausalLM, Tokenizer, TransformerConfig, String, int, float, int, long, float, Consumer<String>) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

stream(CausalLM, Tokenizer, TransformerConfig, String, int, float, int, long, Consumer<String>) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

stream(String, int, float, int, long, float, Consumer<String>) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

stream(String, int, float, int, long, Consumer<String>) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

stream(String, Consumer<String>) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

stream(Function<int[], Tensor>, int, Tokenizer, String, int, float, int, long, float, Consumer<String>) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

stream(Function<int[], Tensor>, int, Tokenizer, String, int, float, int, long, Consumer<String>) - Static method in class io.github.kirstenali.deepj.nn.transformer.TextGenerator

strideHeight() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the strideHeight record component.

strides() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

strideWidth() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the strideWidth record component.

subtract(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

subtract(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BinaryOps

subtract(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.BroadcastOps

subtract(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

subtract(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

subtract(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

subtractBroadcastCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

subtractBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

subtractBroadcastCols(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

subtractBroadcastCols(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

subtractInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

subtractInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

subtractInPlace(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

subtractInPlace(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sum() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sum(int, boolean) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

sum(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ReductionOps

sum(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sum(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sum(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sum(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.autograd.AxisReductionOps

sum(Tensor, int[], boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdReduction

sumAbs() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sumAbs(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sumAbs(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sumAbs(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sumAlongCols() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sumAlongCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sumAlongCols(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sumAlongCols(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sumAlongRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sumAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sumAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sumAlongRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sumAxes(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

sumAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sumAxes(Tensor, int[], boolean) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sumAxes(Tensor, int[], boolean) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

summary() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the summary record component.

sumRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sumRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

sumRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

sumRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

sumTensor() - Method in class io.github.kirstenali.deepj.tensor.Tensor

sumToShape(Tensor, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

SupervisedTraining - Class in io.github.kirstenali.deepj.training

supportsBatchedForward() - Method in interface io.github.kirstenali.deepj.nn.transformer.CausalLM

supportsBatchedForward() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOrigin

supportsCachedGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

supportsCausalAttention(Tensor, Tensor, Tensor, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

supportsCausalAttention(Tensor, Tensor, Tensor, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

supportsCausalAttention(Tensor, Tensor, Tensor, int, int, float) - Method in interface io.github.kirstenali.deepj.tensor.CausalAttentionBackend

supportsCausalAttention(Tensor, Tensor, Tensor, int, int, float) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

supportsIncrementalInference() - Method in interface io.github.kirstenali.deepj.layers.Layer

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.GroupedQueryAttention

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadLatentAttention

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJNovaTransformerBlock

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOrbitTransformerBlock

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJOriginTransformerBlock

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.layers.transformer.blocks.DeepJPrismTransformerBlock

supportsIncrementalInference() - Method in class io.github.kirstenali.deepj.transformer.LoopedTransformer

supportsIncrementalInference() - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

supportsLayouts() - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

supportsNative16Matmul(DType) - Static method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

Whether compact matrix products can use native16 operands with FP32 accumulation.

supportsRollingGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

supportsRollingGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoop

supportsRollingGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNova

supportsRollingGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbit

supportsRollingGeneration() - Method in class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrism

swiGlu(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

swiGlu(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

swiGlu(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

swiGlu(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

swiGluBackward(Tensor, Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

swiGluBackward(Tensor, Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

swiGluBackward(Tensor, Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

SwiGluBackwardResult - Record Class in io.github.kirstenali.deepj.tensor

SwiGluBackwardResult(Tensor, Tensor) - Constructor for record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Creates an instance of a SwiGluBackwardResult record class.

SwiGLULayer - Class in io.github.kirstenali.deepj.layers.transformer

SwiGLULayer(int, int, Random) - Constructor for class io.github.kirstenali.deepj.layers.transformer.SwiGLULayer

SwiGLULayer(int, int, Random, boolean) - Constructor for class io.github.kirstenali.deepj.layers.transformer.SwiGLULayer

SwiGLULayer(int, int, Random, boolean, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.layers.transformer.SwiGLULayer

T

tanh() - Static method in class io.github.kirstenali.deepj.nn.Activation

tanh(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ActivationOps

tanh(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

tanh(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

tanh(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Tanh - Class in io.github.kirstenali.deepj.activations

Tanh() - Constructor for class io.github.kirstenali.deepj.activations.Tanh

tanhActivation() - Method in class io.github.kirstenali.deepj.tensor.Tensor

tanhInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

tanhInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

tanhInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

tanhInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

tensor() - Method in class io.github.kirstenali.deepj.tensor.Tensor

tensor() - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

Tensor - Class in io.github.kirstenali.deepj.tensor

Tensor(int...) - Constructor for class io.github.kirstenali.deepj.tensor.Tensor

Tensor(int, int) - Constructor for class io.github.kirstenali.deepj.tensor.Tensor

Tensor(DType, int...) - Constructor for class io.github.kirstenali.deepj.tensor.Tensor

Tensor(Tensor) - Constructor for class io.github.kirstenali.deepj.tensor.Tensor

TensorAdapters - Class in io.github.kirstenali.deepj.tensor

TensorBackend - Interface in io.github.kirstenali.deepj.tensor

TensorDescriptor - Record Class in io.github.kirstenali.deepj.tensor

Validated logical checkpoint metadata; reading it never allocates tensor values.

TensorDescriptor(DType, int[]) - Constructor for record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Creates an instance of a TensorDescriptor record class.

TensorEncoding - Class in io.github.kirstenali.deepj.tensor

Raw-bit, logical-order tensor payloads; no FLOAT32 shadow is needed for compact storage.

TensorJoin - Class in io.github.kirstenali.deepj.nn

Backend copies plus a slicing autograd rule; never reads tensor values on the host.

TensorManipulations - Class in io.github.kirstenali.deepj.tensor

Shape transformations retain one Tensor system and participate in first-order autograd.

tensors() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns the value of the tensors record component.

TensorScope - Class in io.github.kirstenali.deepj.tensor

Bounds internal view lifetimes, without releasing saved or explicitly retained storage.

TensorStorageAccess - Class in io.github.kirstenali.deepj.tensor

Shared physical storage plumbing; logical Tensor operations remain on the concrete facade.

text() - Method in record class io.github.kirstenali.deepj.chatui.ChatMessage

Returns the value of the text record component.

TextDataset - Class in io.github.kirstenali.deepj.data

TextFileRange - Record Class in io.github.kirstenali.deepj.data

TextFileRange(long, long) - Constructor for record class io.github.kirstenali.deepj.data.TextFileRange

Creates an instance of a TextFileRange record class.

TextGenerator - Class in io.github.kirstenali.deepj.nn.transformer

tied(Tensor, Tensor) - Static method in class io.github.kirstenali.deepj.layers.ProjectionOps

TiedEmbeddingProjection - Class in io.github.kirstenali.deepj.layers

Predicts tokens using the same parameter that supplies the input embeddings.

TiedEmbeddingProjection(Parameter) - Constructor for class io.github.kirstenali.deepj.layers.TiedEmbeddingProjection

to(DType) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

toFloatArray() - Method in class io.github.kirstenali.deepj.tensor.TensorStorageAccess

Returns a logical row-major copy, converting compact storage to FLOAT32 values when necessary.

toIntTargets(Tensor) - Static method in class io.github.kirstenali.deepj.loss.CrossEntropyLoss

tokEmb - Variable in class io.github.kirstenali.deepj.nn.transformer.DecoderOnlyModel

tokenizer() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns the value of the tokenizer record component.

Tokenizer - Interface in io.github.kirstenali.deepj.tokenizers

TOKENIZER_FILE - Static variable in class io.github.kirstenali.deepj.publishing.DeepJModelBundle

TokenizerConfig(int, int) - Constructor for record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Creates an instance of a TokenizerConfig record class.

tokenKeyToId() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns the value of the tokenKeyToId record component.

TokenPair - Record Class in io.github.kirstenali.deepj.tokenizers.bpe

TokenPair(int, int) - Constructor for record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Creates an instance of a TokenPair record class.

tokens() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns the value of the tokens record component.

tokens() - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Returns the value of the tokens record component.

toString() - Method in record class io.github.kirstenali.deepj.chatui.ChatMessage

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.data.Batch

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Example

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.data.TextFileRange

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Architecture

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.FilesConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.FormatAlpacaText.FormatResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.PreparationResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.RecurrentResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.RecurrentState

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.persistence.ModelInspection

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.persistence.ModelLayout

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.persistence.ModelSettings

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.CausalAttentionResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.CrossEntropyResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.NormalizationResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.RmsNormResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.training.EvaluationResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.training.FixedTextValidation.Settings

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.training.TrainingProgress

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.training.TrainingResult

Returns a string representation of this record class.

toString() - Method in record class io.github.kirstenali.deepj.transformer.TransformerStack

Returns a string representation of this record class.

total() - Method in record class io.github.kirstenali.deepj.examples.GenerateFactDataset.GenerationResult

total(Tensor, boolean) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdReduction

totalHeads(Tensor, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadLayout

totalSteps() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns the value of the totalSteps record component.

tracks(Tensor...) - Static method in class io.github.kirstenali.deepj.autograd.Autograd

train() - Method in class io.github.kirstenali.deepj.nn.Module

train(int, int, int, float, Float) - Method in class io.github.kirstenali.deepj.training.Trainer

train(int, int, int, float, Float, int) - Method in class io.github.kirstenali.deepj.training.Trainer

train(int, int, int, float, Float, int, Trainer.StepHook) - Method in class io.github.kirstenali.deepj.training.Trainer

train(int, int, int, float, Float, int, Trainer.StepHook, TrainingProgress) - Method in class io.github.kirstenali.deepj.training.Trainer

train(int, int, int, float, Float, Trainer.StepHook) - Method in class io.github.kirstenali.deepj.training.Trainer

train(String, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

train(String, int, List<String>) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

Trainable - Interface in io.github.kirstenali.deepj.training

TrainDeepJ100M - Class in io.github.kirstenali.deepj.examples

Trains Nova from scratch using Frisky's FineWeb-Edu corpus and tokenizer.

TrainDeepJ90M - Class in io.github.kirstenali.deepj.examples

TrainDeepJLoop100M - Class in io.github.kirstenali.deepj.examples

Runs the pretrain stage using the existing Frisky dataset.

TrainDeepJPrismKnowledge - Class in io.github.kirstenali.deepj.examples

trainer(Layer, LossFunction, ParameterOptimizer, Tensor, Tensor, long) - Static method in class io.github.kirstenali.deepj.nn.NdSupervisedTraining

trainer(Layer, LossFunction, ParameterOptimizer, Tensor, Tensor, long) - Static method in class io.github.kirstenali.deepj.training.SupervisedTraining

trainer(CausalLM, BatchSource, float) - Static method in class io.github.kirstenali.deepj.training.CausalLMTraining

trainer(CausalLM, BatchSource, ParameterOptimizer) - Static method in class io.github.kirstenali.deepj.training.CausalLMTraining

Trainer - Class in io.github.kirstenali.deepj.training

Trainer(Trainer.StepFunction) - Constructor for class io.github.kirstenali.deepj.training.Trainer

Trainer.StepFunction - Interface in io.github.kirstenali.deepj.training

Trainer.StepHook - Interface in io.github.kirstenali.deepj.training

trainFromFile(Path, int, List<String>) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainFromFile(Path, int, List<String>, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

training() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig

Returns the value of the training record component.

training() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Returns the value of the training record component.

Training(int, int, float, float, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Creates an instance of a Training record class.

TrainingCheckpoint - Class in io.github.kirstenali.deepj.persistence

trainingData() - Method in record class io.github.kirstenali.deepj.publishing.ModelCard

Returns the value of the trainingData record component.

TrainingProgress - Record Class in io.github.kirstenali.deepj.training

TrainingProgress(int, float, float) - Constructor for record class io.github.kirstenali.deepj.training.TrainingProgress

Creates an instance of a TrainingProgress record class.

TrainingResult - Record Class in io.github.kirstenali.deepj.training

TrainingResult(int, float, float) - Constructor for record class io.github.kirstenali.deepj.training.TrainingResult

Creates an instance of a TrainingResult record class.

TrainSmallDeepJOrbit - Class in io.github.kirstenali.deepj.examples

TrainSmallDeepJOrbit() - Constructor for class io.github.kirstenali.deepj.examples.TrainSmallDeepJOrbit

TrainSmallDeepJOrigin - Class in io.github.kirstenali.deepj.examples

TrainSmallDeepJOrigin() - Constructor for class io.github.kirstenali.deepj.examples.TrainSmallDeepJOrigin

TrainSmallDeepJOriginWithBPE - Class in io.github.kirstenali.deepj.examples

TrainSmallDeepJOriginWithBPE() - Constructor for class io.github.kirstenali.deepj.examples.TrainSmallDeepJOriginWithBPE

TrainSmallDeepJPrism - Class in io.github.kirstenali.deepj.examples

TrainSmallDeepJPrism() - Constructor for class io.github.kirstenali.deepj.examples.TrainSmallDeepJPrism

trainStep(int) - Method in interface io.github.kirstenali.deepj.training.Trainer.StepFunction

trainStep(int) - Method in class io.github.kirstenali.deepj.training.Trainer

trainTokenizer(String, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainTokenizer(String, int, List<String>) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainTokenizerFromFile(Path, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainTokenizerFromFile(Path, int, List<String>) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainTokenizerWithDefaults(String, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

trainTokenizerWithDefaultsFromFile(Path, int) - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETrainer

transferLayout(Tensor, DType) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

TransformerConfig - Interface in io.github.kirstenali.deepj.nn.transformer

TransformerEncoder - Class in io.github.kirstenali.deepj.nn

Stacked encoder for [batch, time, width] embeddings, with padding masks and masked mean pooling.

TransformerEncoder(TransformerEncoderConfig, Random) - Constructor for class io.github.kirstenali.deepj.nn.TransformerEncoder

TransformerEncoder(TransformerEncoderConfig, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.TransformerEncoder

TransformerEncoderConfig - Record Class in io.github.kirstenali.deepj.nn

Pre-normalized encoder dimensions.

TransformerEncoderConfig(int, int, int, int, float) - Constructor for record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Creates an instance of a TransformerEncoderConfig record class.

TransformerEncoderLayer - Class in io.github.kirstenali.deepj.nn

Pre-normalized, bidirectional encoder block reusing DeepJ linear layers and LayerNorm.

TransformerEncoderLayer(TransformerEncoderConfig, Random, ParameterFactory) - Constructor for class io.github.kirstenali.deepj.nn.TransformerEncoderLayer

TransformerStack - Record Class in io.github.kirstenali.deepj.transformer

TransformerStack(List<Layer>) - Constructor for record class io.github.kirstenali.deepj.transformer.TransformerStack

Creates an instance of a TransformerStack record class.

transformQueryKey(Tensor, int) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

transformQueryKey(Tensor, int) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.RoPEMultiHeadSelfAttention

transformQueryKey(Tensor, int, int) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

transformQueryKey(Tensor, int, int) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.RoPEMultiHeadSelfAttention

transformQueryKey(Tensor, int, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.MultiHeadSelfAttention

transformQueryKey(Tensor, int, InferenceContext) - Method in class io.github.kirstenali.deepj.layers.transformer.attention.RoPEMultiHeadSelfAttention

transpose() - Method in class io.github.kirstenali.deepj.tensor.Tensor

transpose(int, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

transpose(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.UnaryOps

transpose(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

transpose(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

transpose(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

transpose(Tensor, int, int) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

transposeLeft() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the transposeLeft record component.

transposeRight() - Method in record class io.github.kirstenali.deepj.tensor.PackedMatmulShape

Returns the value of the transposeRight record component.

U

unary(Tensor, CpuNdElementwise.Unary) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

unaryInto(Tensor, Tensor, CpuNdElementwise.Unary) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdElementwise

UnaryOps - Class in io.github.kirstenali.deepj.autograd

unpackF32(float[], int, int) - Static method in class io.github.kirstenali.deepj.tensor.TensorAdapters

unpackF32Into(float[], Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorAdapters

unsqueeze(int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

unsqueeze(Tensor, int) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

unsqueeze(Tensor, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

update(Tensor, Tensor, Tensor, Tensor, float, float, float, float, float, float, float) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdAdamW

upGradient() - Method in record class io.github.kirstenali.deepj.tensor.SwiGluBackwardResult

Returns the value of the upGradient record component.

uploadBuffer(int, float[]) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

uploadBuffer16(int, short[]) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

user() - Method in record class io.github.kirstenali.deepj.chatui.ChatMessage

Returns the value of the user record component.

usesDeviceStorage() - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

usesDeviceStorage() - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Whether backend copies reside on a device rather than in ordinary host storage.

V

validateCommon(int, int, int, int, int, int, float) - Static method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

validateRotaryHeadDimension(int, int) - Static method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

validateState(List<Parameter>, AdamW.State) - Method in class io.github.kirstenali.deepj.optimisers.AdamW

Checks a prospective checkpoint without changing parameters, counters or optimizer buffers.

validation() - Method in record class io.github.kirstenali.deepj.examples.PrepareKnowledgeCorpus.SplitResult

Returns the value of the validation record component.

value - Variable in class io.github.kirstenali.deepj.optimisers.Parameter

value() - Method in record class io.github.kirstenali.deepj.tensor.AttentionGradients

Returns the value of the value record component.

valueOf(String) - Static method in enum class io.github.kirstenali.deepj.nn.Pool2D.Reduction

Returns the enum constant of this class with the specified name.

valueOf(String) - Static method in enum class io.github.kirstenali.deepj.nn.transformer.ContextWindowMode

Returns the enum constant of this class with the specified name.

valueOf(String) - Static method in enum class io.github.kirstenali.deepj.nn.transformer.PromptFormat

Returns the enum constant of this class with the specified name.

valueOf(String) - Static method in enum class io.github.kirstenali.deepj.tensor.DType

Returns the enum constant of this class with the specified name.

values() - Static method in enum class io.github.kirstenali.deepj.nn.Pool2D.Reduction

Returns an array containing the constants of this enum class, in the order they are declared.

values() - Static method in enum class io.github.kirstenali.deepj.nn.transformer.ContextWindowMode

Returns an array containing the constants of this enum class, in the order they are declared.

values() - Static method in enum class io.github.kirstenali.deepj.nn.transformer.PromptFormat

Returns an array containing the constants of this enum class, in the order they are declared.

values() - Static method in enum class io.github.kirstenali.deepj.tensor.DType

Returns an array containing the constants of this enum class, in the order they are declared.

variance(int, boolean) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

Population variance; use the correction overload for a different denominator.

variance(int, boolean, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

variance(Tensor) - Static method in class io.github.kirstenali.deepj.autograd.ReductionOps

variance(Tensor, int[], boolean, int) - Static method in class io.github.kirstenali.deepj.autograd.AxisReductionOps

variance(Tensor, int[], boolean, int) - Static method in class io.github.kirstenali.deepj.tensor.CpuNdStatistics

varianceAlongRows() - Method in class io.github.kirstenali.deepj.tensor.Tensor

varianceAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

varianceAlongRows(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

varianceAlongRows(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

varianceAxes(int[], boolean, int) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

varianceAxes(Tensor, int[], boolean, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

varianceAxes(Tensor, int[], boolean, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

varianceAxes(Tensor, int[], boolean, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

version() - Method in class io.github.kirstenali.deepj.autograd.GradientState

view(int...) - Method in class io.github.kirstenali.deepj.tensor.TensorManipulations

view(Tensor, int...) - Static method in class io.github.kirstenali.deepj.autograd.ViewOps

view(Tensor, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

ViewOps - Class in io.github.kirstenali.deepj.autograd

VOCAB_SIZE - Static variable in class io.github.kirstenali.deepj.tokenizers.ByteTokenizer

vocabSize() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.TokenizerConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in record class io.github.kirstenali.deepj.nn.transformer.loop.DeepJLoopConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in record class io.github.kirstenali.deepj.nn.transformer.nova.DeepJNovaConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in record class io.github.kirstenali.deepj.nn.transformer.orbit.DeepJOrbitConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in record class io.github.kirstenali.deepj.nn.transformer.origin.DeepJOriginConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in record class io.github.kirstenali.deepj.nn.transformer.prism.DeepJPrismConfig

Returns the value of the vocabSize record component.

vocabSize() - Method in interface io.github.kirstenali.deepj.nn.transformer.TransformerConfig

vocabSize() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPEModel

vocabSize() - Method in class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

vocabSize() - Method in class io.github.kirstenali.deepj.tokenizers.ByteTokenizer

vocabSize() - Method in interface io.github.kirstenali.deepj.tokenizers.Tokenizer

W

warmupSteps() - Method in record class io.github.kirstenali.deepj.examples.DeepJPrismTrainingConfig.Training

Returns the value of the warmupSteps record component.

warmupSteps() - Method in record class io.github.kirstenali.deepj.training.CosineLearningRateSchedule

Returns the value of the warmupSteps record component.

weight() - Method in record class io.github.kirstenali.deepj.data.ResponseOnlyTextDataset.Source

Returns the value of the weight record component.

weight() - Method in class io.github.kirstenali.deepj.layers.Linear

weight() - Method in class io.github.kirstenali.deepj.nn.Conv1D

weight() - Method in class io.github.kirstenali.deepj.nn.Conv2D

weight() - Method in class io.github.kirstenali.deepj.transformer.embeddings.Embedding

weightDecay() - Method in record class io.github.kirstenali.deepj.optimisers.AdamW.State

Returns the value of the weightDecay record component.

width() - Method in record class io.github.kirstenali.deepj.nn.TransformerEncoderConfig

Returns the value of the width record component.

width() - Method in record class io.github.kirstenali.deepj.nn.Window2D

Returns the value of the width record component.

width(Tensor) - Static method in class io.github.kirstenali.deepj.tensor.NormalizationChecks

window() - Method in class io.github.kirstenali.deepj.nn.Conv2D

Window2D - Record Class in io.github.kirstenali.deepj.nn

Floor-mode spatial window with symmetric zero padding and optional dilation.

Window2D(int, int) - Constructor for record class io.github.kirstenali.deepj.nn.Window2D

Window2D(int, int, int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.nn.Window2D

Creates an instance of a Window2D record class.

withContextWindow(ContextWindowMode) - Method in class io.github.kirstenali.deepj.nn.transformer.DeepJChat

Returns a view sharing this model and tokenizer, with independent request-scoped caches.

write(DataOutput) - Method in record class io.github.kirstenali.deepj.tensor.TensorDescriptor

write(DataOutput, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

writeHeadCache(Tensor, Tensor, int, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

writeHeadCache(Tensor, Tensor, int, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

writeLayouts(TensorBackend, Tensor, Tensor, int, int) - Static method in class io.github.kirstenali.deepj.tensor.HeadCacheOps

writeValues(DataOutput, Tensor) - Static method in class io.github.kirstenali.deepj.tensor.TensorEncoding

X

x() - Method in record class io.github.kirstenali.deepj.data.Batch

Returns the value of the x record component.

Y

y() - Method in record class io.github.kirstenali.deepj.data.Batch

Returns the value of the y record component.

Z

zeroBuffer(int, int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

Initializes the whole buffer with positive zero; custom runtimes retain a portable fallback.

zeroBuffer(int, int, DType) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

zeroGrad() - Method in class io.github.kirstenali.deepj.optimisers.Parameter

zeroGrad() - Method in class io.github.kirstenali.deepj.tensor.Tensor

zeroGrad() - Method in interface io.github.kirstenali.deepj.training.Trainable

zeros(int, int) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

zeros(int, int) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

zeros(int, int) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

zeros(int, int) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Backend-owned positive-zero storage; ordinary Tensor constructors stay CPU-backed.

zeros(DType, int...) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

zeros(DType, int...) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

zeros(DType, int...) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

zeros(DType, int...) - Static method in class io.github.kirstenali.deepj.tensor.TensorManipulations

zeros(String, int, int) - Method in class io.github.kirstenali.deepj.optimisers.ParameterFactory

zeroToDevice(DType, int...) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

A B C D E F G H I K L M N O P Q R S T U V W X Y Z
All Classes and Interfaces|All Packages|Constant Field Values