Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/crabml/crabml
/ types & classes
Types & classes
96 in github.com/crabml/crabml
⨍
Functions
598
◇
Types & classes
96
Enum
Activation
crabml-llama2/src/llama2.rs:21
Class
ArithmeticPushConstants
crabml-vulkan/src/push_constants.rs:5
Class
BatchMatmuPushConstants
crabml-vulkan/src/push_constants.rs:70
Class
BatchMatmulMeta
crabml-wgpu/src/meta.rs:23
Class
BlockQ2K
crabml-core/src/cpu/buf/buf_q2_k.rs:19
Class
BlockQ3K
crabml-core/src/cpu/buf/buf_q3_k.rs:21
Class
BlockQ4K
crabml-core/src/cpu/buf/buf_q4_k.rs:16
Class
BlockQ4_0
crabml-core/src/cpu/buf/buf_q4_0.rs:12
Class
BlockQ4_1
crabml-core/src/cpu/buf/buf_q4_1.rs:12
Class
BlockQ5K
crabml-core/src/cpu/buf/buf_q5_k.rs:15
Class
BlockQ5_0
crabml-core/src/cpu/buf/buf_q5_0.rs:15
Class
BlockQ5_1
crabml-core/src/cpu/buf/buf_q5_1.rs:12
Class
BlockQ6K
crabml-core/src/cpu/buf/buf_q6_k.rs:13
Class
BlockQ8K
crabml-core/src/cpu/buf/buf_q8_k.rs:8
Class
BlockQ8_0
crabml-core/src/cpu/buf/buf_q8_0.rs:10
Class
BlockQ8_1
crabml-core/src/cpu/buf/buf_q8_1.rs:75
Enum
ChatTemplate
crabml-llama2/src/chat.rs:191
Class
CommandArgs
crabml-cli/src/main.rs:30
Class
ConcatenateMeta
crabml-wgpu/src/meta.rs:45
Class
ConcatenatePushConstants
crabml-vulkan/src/push_constants.rs:48
Class
ContiguousMeta
crabml-wgpu/src/meta.rs:57
Class
ContiguousPushConstants
crabml-vulkan/src/push_constants.rs:39
Class
CpuLlamaModel
crabml-llama2/src/model.rs:102
Class
CpuLlamaModelLoader
crabml-llama2/src/model.rs:139
Class
CpuTensor
crabml-core/src/cpu/cpu_tensor.rs:16
Enum
CpuTensorBuf
crabml-core/src/cpu/buf/api.rs:32
Class
CpuTensorDevice
crabml-core/src/cpu/cpu_device.rs:51
Class
CpuTensorDeviceOptions
crabml-core/src/cpu/cpu_device.rs:13
Enum
DeviceType
crabml-cli/src/main.rs:67
Class
Error
crabml-core/src/error.rs:36
Enum
ErrorKind
crabml-core/src/error.rs:6
Enum
GGMLType
crabml-core/src/gguf.rs:86
Class
GGUFBufReader
crabml-core/src/gguf.rs:247
Class
GGUFFile
crabml-core/src/gguf.rs:689
Class
GGUFFileLoader
crabml-core/src/gguf.rs:789
Class
GGUFHeader
crabml-core/src/gguf.rs:492
Class
GGUFMetadata
crabml-core/src/gguf.rs:427
Enum
GGUFMetadataArray
crabml-core/src/gguf.rs:231
Class
GGUFMetadataReader
crabml-core/src/gguf.rs:284
Enum
GGUFMetadataValue
crabml-core/src/gguf.rs:194
Enum
GGUFMetadataValueType
crabml-core/src/gguf.rs:149
Class
GGUFOnDiskTensorInfo
crabml-core/src/gguf.rs:613
Class
GGUFTensorInfo
crabml-core/src/gguf.rs:649
Enum
GGUFVersion
crabml-core/src/gguf.rs:68
Class
Gpt2Tokenizer
crabml-core/src/tokenizer/tokenizer_gpt2.rs:8
Class
GpuLlamaModel
crabml-llama2/src/model.rs:637
Class
Llama2Chat
crabml-llama2/src/chat.rs:7
Class
Llama2ChatReplyIterator
each dialog has a start mark and an end mark. The chat iterator will take the generation result from the model and concatenate them until got the end
crabml-llama2/src/chat.rs:73
Class
Llama2ChatReplyIteratorStats
crabml-llama2/src/chat.rs:64
Class
Llama2Runner
crabml-llama2/src/llama2.rs:26
Class
Llama2Sampler
crabml-llama2/src/sampler.rs:10
Class
LlamaConfig
crabml-llama2/src/model.rs:30
Interface
LlamaModel
crabml-llama2/src/model.rs:86
Class
LlamaTokenizer
crabml-core/src/tokenizer/tokenizer_llama.rs:6
Class
LlamaWeights
crabml-llama2/src/model.rs:55
Enum
MarkMatchState
crabml-llama2/src/chat.rs:133
Class
MarkMatcher
the model response might ends with stop mark like "<end_of_turn>", please note these stop mark are NOT a single token. we'd use a simple state machine
crabml-llama2/src/chat.rs:127
Class
MatmulMeta
crabml-wgpu/src/meta.rs:13
Class
MatmulPushConstants
crabml-vulkan/src/push_constants.rs:61
Enum
ModelArchitecture
crabml-llama2/src/model.rs:22
Class
QuantBufQ2K
crabml-core/src/cpu/buf/buf_q2_k.rs:84
Class
QuantBufQ3K
crabml-core/src/cpu/buf/buf_q3_k.rs:99
Class
QuantBufQ4K
crabml-core/src/cpu/buf/buf_q4_k.rs:51
Class
QuantBufQ4_0
crabml-core/src/cpu/buf/buf_q4_0.rs:31
Class
QuantBufQ4_1
crabml-core/src/cpu/buf/buf_q4_1.rs:34
Class
QuantBufQ5K
crabml-core/src/cpu/buf/buf_q5_k.rs:64
Class
QuantBufQ5_0
crabml-core/src/cpu/buf/buf_q5_0.rs:40
Class
QuantBufQ5_1
crabml-core/src/cpu/buf/buf_q5_1.rs:39
Class
QuantBufQ6K
crabml-core/src/cpu/buf/buf_q6_k.rs:51
Class
QuantBufQ8K
crabml-core/src/cpu/buf/buf_q8_k.rs:24
Class
QuantBufQ8_0
crabml-core/src/cpu/buf/buf_q8_0.rs:27
Class
QuantBufQ8_1
crabml-core/src/cpu/buf/buf_q8_1.rs:16
Class
RmsNormMeta
crabml-wgpu/src/meta.rs:3
Class
RmsNormPushConstants
crabml-vulkan/src/push_constants.rs:21
Class
RopeMeta
crabml-wgpu/src/meta.rs:34
Enum
RopeMode
crabml-core/src/tensor/api.rs:6
Class
RopePushConstants
crabml-vulkan/src/push_constants.rs:29
Class
Scope
crabml-core/src/cpu/thread_pool.rs:73
Class
SoftmaxPushConstants
crabml-vulkan/src/push_constants.rs:14
Interface
Tensor
crabml-core/src/tensor/api.rs:11
Class
TensorMetrics
crabml-core/src/tensor/metrics.rs:6
Class
TensorStrider
crabml-core/src/tensor/strider.rs:6
Class
ThreadPool
crabml-core/src/cpu/thread_pool.rs:13
Class
TimeMetric
crabml-core/src/tensor/metrics.rs:146
Class
TimeMetricGuard
crabml-core/src/tensor/metrics.rs:150
Class
Tokenizer
crabml-core/src/tokenizer/mod.rs:13
Enum
TokenizerInner
crabml-core/src/tokenizer/mod.rs:19
Enum
TokenizerKind
crabml-core/src/tokenizer/mod.rs:25
Class
Utf8Buf
crabml-core/src/tokenizer/mod.rs:114
Class
VulkanTensor
crabml-vulkan/src/vulkan_tensor.rs:22
Class
VulkanTensorDevice
crabml-vulkan/src/vulkan_device.rs:76
Class
VulkanTensorDeviceInner
Vulkan has a lot of boilerplate code, put it them a stand-alone struct to deal with it. I believe the complexity of Vulkan mostly comes from the fact
crabml-vulkan/src/vulkan_device.rs:210
Class
VulkanTensorDeviceOptions
crabml-vulkan/src/vulkan_device.rs:45
Class
WgpuTensor
crabml-wgpu/src/wgpu_tensor.rs:21
Class
WgpuTensorDevice
crabml-wgpu/src/wgpu_device.rs:40
Class
WgpuTensorDeviceOptions
crabml-wgpu/src/wgpu_device.rs:9