List of all items

Structs

• autograd::TVar
• autograd::Tape
• backend::wgpu::WgpuContext
• models::gpt2::AttnStep
• models::gpt2::Gpt2
• models::gpt2::Gpt2Config
• models::gpt2::KvCache
• models::gpt2::LayerDetail
• models::gpt2::StepTrace
• nn::Embedding
• nn::LayerNorm
• nn::Linear
• ops::MatmulSpec
• optim::AdamW
• optim::AdamWOpts
• shape::Shape
• tensor::Tensor
• tensor::WgpuStorage
• tokenizer::Gpt2Tokenizer
• tokenizer::char::CharTokenizer

Enums

• device::Device
• dtype::DType
• error::ForgeError
• models::gpt2::Sampling
• tensor::CpuStorage
• tensor::Storage
• tokenizer::AnyTokenizer

Traits

• tokenizer::Tokenizer

Functions

• backend::cpu::adamw
• backend::cpu::add
• backend::cpu::ce_bwd
• backend::cpu::dropout
• backend::cpu::embedding
• backend::cpu::gather_nll
• backend::cpu::gelu
• backend::cpu::gelu_bwd
• backend::cpu::kv_append
• backend::cpu::layernorm
• backend::cpu::layernorm_bwd_dparams
• backend::cpu::layernorm_bwd_dx
• backend::cpu::matmul
• backend::cpu::merge_heads
• backend::cpu::pcg_hash
• backend::cpu::scatter_add_rows
• backend::cpu::softmax
• backend::cpu::softmax_bwd
• backend::cpu::split_heads
• backend::cpu::sum_rows
• backend::cpu::unmerge_heads
• backend::cpu::unsplit_head
• backend::wgpu::linear_grid
• backend::wgpu::ops::adamw
• backend::wgpu::ops::add
• backend::wgpu::ops::alloc_storage
• backend::wgpu::ops::ce_bwd
• backend::wgpu::ops::dropout
• backend::wgpu::ops::embedding
• backend::wgpu::ops::gather_nll
• backend::wgpu::ops::gelu
• backend::wgpu::ops::gelu_bwd
• backend::wgpu::ops::kv_append
• backend::wgpu::ops::layernorm
• backend::wgpu::ops::layernorm_bwd
• backend::wgpu::ops::matmul
• backend::wgpu::ops::matmul_into
• backend::wgpu::ops::merge_heads
• backend::wgpu::ops::scale
• backend::wgpu::ops::scatter_add_rows
• backend::wgpu::ops::softmax
• backend::wgpu::ops::softmax_bwd
• backend::wgpu::ops::split_heads
• backend::wgpu::ops::sum_rows
• backend::wgpu::ops::sumsq_partials
• backend::wgpu::ops::unmerge_heads
• backend::wgpu::ops::unsplit_head
• ops::adamw
• ops::add
• ops::ce_bwd
• ops::dropout
• ops::embedding
• ops::embedding_chunked
• ops::gather_nll
• ops::gelu
• ops::gelu_bwd
• ops::kv_append
• ops::layernorm
• ops::layernorm_bwd
• ops::matmul
• ops::matmul_chunked_transb
• ops::merge_heads
• ops::scale
• ops::scatter_add_rows
• ops::softmax
• ops::softmax_bwd
• ops::split_heads
• ops::sum_rows
• ops::sumsq
• ops::unmerge_heads
• ops::unsplit_head
• serialization::save_safetensors

Type Aliases

• error::Result

Constants

• backend::wgpu::OFFSET_ALIGN_BYTES