Module: Toy::LLM

Defined in:

lib/toy/llm/recipes/lora.rb,
lib/toy/llm/classify_batch.rb,
lib/toy/llm/primitives/gqa.rb,
lib/toy/llm/recipe_options.rb,
lib/toy/llm/training_batch.rb,
lib/toy/llm/primitives/rope.rb,
lib/toy/llm/archs/llama_arch.rb,
lib/toy/llm/recipes/vit_tiny.rb,
lib/toy/llm/primitives/swiglu.rb,
lib/toy/llm/recipes/lora_cuda.rb,
lib/toy/llm/recipes/lora_metal.rb,
lib/toy/llm/recipes/warm_start.rb,
lib/toy/llm/primitives/gqa_cuda.rb,
lib/toy/llm/primitives/rms_norm.rb,
lib/toy/llm/primitives/gqa_metal.rb,
lib/toy/llm/primitives/rope_cuda.rb,
lib/toy/llm/recipes/from_scratch.rb,
lib/toy/llm/archs/llama_arch_cuda.rb,
lib/toy/llm/primitives/rope_metal.rb,
lib/toy/llm/archs/llama_arch_metal.rb,
lib/toy/llm/engine/gpt2_seq_engine.rb,
lib/toy/llm/engine/vit_tiny_engine.rb,
lib/toy/llm/primitives/swiglu_cuda.rb,
lib/toy/llm/engine/llama_seq_engine.rb,
lib/toy/llm/primitives/swiglu_metal.rb,
lib/toy/llm/recipes/warm_start_cuda.rb,
lib/toy/llm/blocks/transformer_block.rb,
lib/toy/llm/primitives/rms_norm_cuda.rb,
lib/toy/llm/recipes/warm_start_metal.rb,
lib/toy/llm/primitives/rms_norm_metal.rb,
lib/toy/llm/recipes/from_scratch_cuda.rb,
lib/toy/llm/recipes/from_scratch_metal.rb,
lib/toy/llm/engine/gpt2_seq_engine_cuda.rb,
lib/toy/llm/engine/gpt2_seq_engine_metal.rb,
lib/toy/llm/engine/llama_seq_engine_cuda.rb,
lib/toy/llm/blocks/transformer_block_cuda.rb,
lib/toy/llm/engine/llama_seq_engine_metal.rb,
lib/toy/llm/blocks/transformer_block_metal.rb

Defined Under Namespace

Modules: Archs, Blocks, Engine, Primitives, Recipes Classes: ClassifyBatch, RecipeOptions, TrainingBatch