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

C

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

Wiring helpers for causal language model training.

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Collects GPU operations lazily and flushes them as a single command buffer.

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

Create a ComputeGraph backed by the given GPU runtime.

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

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

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

True if GPU has newer data than Tensor.data (CPU cache is stale).

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

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

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

Create a Tensor backed by a GpuBuffer.

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

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

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 - Class in io.github.kirstenali.deepj.loss

Cross-entropy loss with integer class targets.

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

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