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

F

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

Activation used inside the FFN.

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

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

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

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

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

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

Flush all recorded ops to the GPU as a single command buffer.

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

Execute a batch of ops encoded as a flat int[] command stream, all in one command buffer submission.

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

Flexible fully-connected neural network (MLP) built from Linear projections.

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

Build an MLP of the form: Linear -> act -> Linear -> act -> ...

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

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 class io.github.kirstenali.deepj.models.gpt.GPTModel

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

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.Softmax

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

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

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.transformer.LayerNorm1D

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

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

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

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

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

Builds a [n x 1] Tensor from int[] targets.

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