Index

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

M

main(String[]) - Static method in class io.github.kirstenali.deepj.chatui.ChatApp

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainClassicFNN

main(String[]) - Static method in class io.github.kirstenali.deepj.examples.TrainSmallGPT

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

Ensures this tensor's data[][] is up to date with any pending GPU computation.

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

Materialize a tensor: flush pending ops if needed, then download GPU data to CPU.

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

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

Materialize a tensor: flush any pending GPU computation and download the result to the tensor's CPU data[][].

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

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

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

maxSeqLen() - Method in record class io.github.kirstenali.deepj.models.gpt.GPTConfig

Returns the value of the maxSeqLen record component.

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

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.

MetalBackend - Class in io.github.kirstenali.deepj.tensor.metal

MetalBackend() - Constructor for class io.github.kirstenali.deepj.tensor.metal.MetalBackend

model() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.BPETokenizer

Returns the value of the model record component.

ModelSerializer - Class in io.github.kirstenali.deepj.persistence

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

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

MultiHeadSelfAttention - Class in io.github.kirstenali.deepj.layers.transformer

Multi-head causal self-attention for a single sequence (no batch dimension).

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

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

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

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 interface io.github.kirstenali.deepj.tensor.TensorBackend

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

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

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

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

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

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

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

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