Module: Ignis::AI

Defined in:

lib/nnw/ai/loss.rb,
lib/nnw/ai/server.rb,
lib/nnw/ai/trainer.rb,
lib/nnw/ai/kv_cache.rb,
lib/nnw/ai/inference.rb,
lib/nnw/ai/nn/linear.rb,
lib/nnw/ai/nn/module.rb,
lib/nnw/ai/optim/sgd.rb,
lib/nnw/ai/tokenizer.rb,
lib/nnw/ai/nn/dropout.rb,
lib/nnw/ai/optim/adam.rb,
lib/nnw/ai/optim/base.rb,
lib/nnw/ai/weight_map.rb,
lib/nnw/ai/gpt2_loader.rb,
lib/nnw/ai/nn/rms_norm.rb,
lib/nnw/ai/optim/adamw.rb,
lib/nnw/ai/safetensors.rb,
lib/nnw/ai/llama_loader.rb,
lib/nnw/ai/nn/embedding.rb,
lib/nnw/ai/nn/layer_norm.rb,
lib/nnw/ai/nn/sequential.rb,
lib/nnw/ai/transformer/block.rb,
lib/nnw/ai/transformer/model.rb,
lib/nnw/ai/optim/lr_scheduler.rb,
lib/nnw/ai/transformer/modern.rb,
lib/nnw/ai/transformer/swiglu.rb,
lib/nnw/ai/transformer/attention.rb,
lib/nnw/ai/transformer/feed_forward.rb

Defined Under Namespace

Modules: GPT2Loader, LlamaLoader, Loss, NN, Optim, Safetensors, Transformer, WeightMap Classes: BatchProcessor, DataLoader, KVCache, Server, TextGenerator, Tokenizer, Trainer