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

G

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

Gaussian Error Linear Unit (GELU), using the tanh approximation popularized by GPT-2.

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

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

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

generate(GPTModel, Tokenizer, GPTConfig, String, int, double, int, long) - Static method in class io.github.kirstenali.deepj.models.gpt.TextGenerator

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

generate(String, int, double, int, long) - Method in class io.github.kirstenali.deepj.chatui.GPTChatService

get(int, int) - Method in class io.github.kirstenali.deepj.tensor.Tensor

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

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

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

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

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

Get the GPU handle (used by GPU backends).

getLoadedModelName() - Method in interface io.github.kirstenali.deepj.chatui.ChatService

getLoadedModelName() - Method in class io.github.kirstenali.deepj.chatui.GPTChatService

getNumThreads() - Static method in class io.github.kirstenali.deepj.concurrent.DeepJExecutor

Current number of threads used by DeepJ.

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

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

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

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

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

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

GPTConfig - Record Class in io.github.kirstenali.deepj.models.gpt

GPTConfig(int, int, int, int, int, int) - Constructor for record class io.github.kirstenali.deepj.models.gpt.GPTConfig

GPTConfig(int, int, int, int, int, int, double, double) - Constructor for record class io.github.kirstenali.deepj.models.gpt.GPTConfig

Creates an instance of a GPTConfig record class.

GPTModel - Class in io.github.kirstenali.deepj.models.gpt

Minimal GPT-style decoder-only transformer for educational/training use.

GPTModel(GPTConfig, long) - Constructor for class io.github.kirstenali.deepj.models.gpt.GPTModel

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

Handle to a GPU-resident float buffer managed by a ComputeGraph.

GpuBuffer(int, int, int, boolean) - Constructor for class io.github.kirstenali.deepj.tensor.GpuBuffer

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

Abstraction over a GPU compute runtime (Metal, CUDA, Vulkan, etc.).

grad - Variable in class io.github.kirstenali.deepj.optimisers.Parameter

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

Returns the value of the gradClipNorm record component.

gradClipNorm() - Method in class io.github.kirstenali.deepj.models.gpt.GPTModel

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

Convenience helper: gradient w.r.t.

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

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