Optimize transformer attention, the calculation that relates tokens to one another, using Flash Attention. The skill guides PyTorch integration, GPU memory reduction, performance measurement, and output comparisons for training or inference with long sequences.
- Reducing GPU memory use with long sequences
- Integrating Flash Attention into PyTorch models
- Comparing runtime and outputs before and after changes
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Prepare large text collections for AI models with fast tokenizers, which split text into model inputs. Train custom vocabularies, trace tokens to original text positions, adjust input lengths, and integrate the results with transformers.
- Tokenizing large text collections
- Training a custom vocabulary
- Mapping extracted names or answers back to original text
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guides LLM fine-tuning with Axolotl, from YAML configuration and dataset formats to distributed training across GPUs. Covers LoRA and QLoRA, methods that train on preferences, multimodal support, and troubleshooting using documentation references and code examples.
- Prepare YAML settings for LLM fine-tuning
- Explore LoRA, QLoRA, and preference training options
- Check dataset formats and input handling
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guide large language model training with OpenRLHF using human preference feedback. Covers training method selection, distributed GPU setup, reward model training, and memory tuning, with examples for PPO, GRPO, and DPO.
- Choosing a training method for human preference alignment
- Setting up distributed training across multiple GPUs
- Training a reward model before PPO
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Explore training workflows that reduce harmful model responses using a written set of principles. The skill covers self-critique, revised training examples, and RLAIF, reinforcement learning from AI feedback, with implementation examples and hardware guidance.
- Designing principles for model safety training
- Creating training examples through self-critique and revision
- Planning reinforcement learning with AI-rated responses
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Use LitGPT to explore language model implementations and train models on custom data. Covers pretrained model loading, LoRA fine-tuning that updates a small set of parameters, pretraining, and preparing trained models for deployment.
- Studying language model architecture through readable code
- Fine-tuning a model on custom instruction data
- Configuring LoRA training for a single GPU
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Configure large language model pretraining with torchtitan across multiple GPUs. Covers ways to split training work, launch jobs on computing clusters, enable Float8 numerical processing, and save or resume training through distributed checkpoints.
- Pretraining Llama 3.1 8B on eight GPUs
- Configuring multi-node training with SLURM
- Enabling Float8 training on H100 GPUs
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guides you through setting up RWKV, generating text, processing long documents in chunks, and fine-tuning models. Explains how to carry context between steps and compare memory use with Transformer models.
- Build streaming text generation
- Process long documents in chunks
- Plan RWKV fine-tuning
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Build safety checks for LLM prompts and responses using Meta's LlamaGuard. This skill covers content classification, batch moderation, serving a moderation API, and connecting checks to NeMo Guardrails.
- Screening user prompts before sending them to an LLM
- Checking generated responses before displaying them
- Building an API for content safety checks
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guide language model training with TRL, from learning to follow instructions to matching human preferences. Covers method selection, training data, reward models that score responses, evaluation, and adjustments for memory limits or unstable training.
- Train a model on instruction and response examples
- Align responses using preferred and rejected answer pairs
- Build a training pipeline using human feedback
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Assess language models with over 60 academic benchmark tests using consistent prompts and scoring. Compare models, track performance during training, and organize research results with support for HuggingFace models, vLLM, and APIs.
- Compare models on standard knowledge and reasoning tests
- Track model performance across training checkpoints
- Organize evaluation results for research reports
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Build data processing pipelines for machine learning, from preprocessing to batch predictions. Uses Ray Data to handle datasets larger than memory, distribute work across CPUs and GPUs, and feed data into training frameworks.
- Preprocessing datasets larger than memory
- Running batch predictions across multiple GPUs
- Preparing image and tabular data for training
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guides reinforcement learning, where models learn from rewards, using torchforge and PyTorch. Covers reward design, custom loss functions, and distributed training so researchers can experiment with algorithms while keeping infrastructure concerns separate.
- Train mathematical reasoning models with GRPO
- Experiment with custom loss functions
- Scale training across multiple GPUs or nodes
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Implement Mamba, a model architecture that processes sequences through an internal state with computation scaling linearly with input length. Covers pretrained text generation, Mamba-1 and Mamba-2 configuration, speed comparisons with Transformers, and GPU memory troubleshooting.
- Explore models for long sequences or streaming applications
- Generate text with pretrained Mamba models
- Compare Mamba-1 and Mamba-2 configurations
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Convert language models to GGUF and reduce their size through quantization, which stores model weights at lower precision. Guides cover compression choices, local inference on CPUs and Apple Silicon, and integration with tools such as Ollama.
- Running a language model on a CPU-only computer
- Choosing model compression for available memory
- Configuring Metal acceleration on Apple Silicon
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Learn how GPT language models work by preparing text, training a small model, and generating samples. Covers CPU experiments, custom datasets, further training of pretrained GPT-2 models, and reproducing GPT-2 with multiple GPUs.
- Train a small model on Shakespeare using a CPU
- Prepare custom text for model training
- Further train a pretrained GPT-2 model
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Evaluate code generation models with HumanEval, MBPP, and multilingual benchmarks. The skill guides test execution and model comparisons using pass@k, an estimate of whether at least one of k generated solutions passes the tests.
- Compare code generation models on shared benchmarks
- Measure code generation performance across languages
- Evaluate instruction-tuned coding models
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Add PyTorch FSDP2 to training scripts so model state can be split across GPUs. Covers distributed initialization, mixed precision, CPU offloading, and checkpoint saving and loading for models that exceed a single GPU's memory.
- Training a model that exceeds one GPU's memory
- Adding FSDP2 to an existing training script
- Configuring mixed precision and CPU offloading
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Fine-tune language models with TRL using GRPO, a method that compares scored responses to the same prompt. Covers training data, scoring functions, GPU-conscious settings, and monitoring for tasks with measurable correctness or required output formats.
- Train models to follow JSON or XML output formats
- Reward correct answers in math or coding tasks
- Balance correctness, format, and style during training
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guides PyTorch model training and fine-tuning, from choosing architectures to configuring optimizers and learning rates. Provides troubleshooting steps for unstable loss and memory shortages, plus methods for improving GPU throughput and comparing experiments.
- Choose model architectures for the available data type and scale
- Configure optimizers and learning-rate schedules
- Investigate loss spikes, NaNs, and GPU memory shortages
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Run machine learning workloads on Modal's cloud GPUs without managing servers. Guides Python configuration for training, inference, batch processing, and API deployment, including automatic scaling, persistent storage, scheduled jobs, and troubleshooting.
- Running GPU inference without managing servers
- Deploying a model as an automatically scaling API
- Processing large batches in parallel
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Evaluate language models on knowledge, math, coding, safety, and image understanding benchmarks. The skill guides reproducible runs in containers using local Docker, Slurm clusters, or cloud execution, with shared configurations for model comparison and result export.
- Comparing models on the same benchmark tasks
- Running a small evaluation sample before a full benchmark
- Running large evaluations on a Slurm cluster
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Guides setup and operation of Lambda Labs GPU instances for machine learning training and inference. Covers SSH access, persistent storage, API automation, and multi-GPU or multi-node training, including how to retain checkpoints and terminate idle instances.
- Setting up a GPU instance for LLM fine-tuning
- Keeping datasets and checkpoints between sessions
- Automating instance launch and termination through the API
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026
Apply HQQ quantization, which stores language model weights using fewer bits without calibration examples. Covers precision settings, inference backends, saving compressed models, deployment with HuggingFace Transformers or vLLM, and further training with LoRA.
- Compressing a model without calibration data
- Comparing 4-bit and 2-bit quantization settings
- Running quantized models with HuggingFace Transformers or vLLM
Orchestra-Research/AI-Research-SKILLs☆ 13.8KUpdated Oct 11, 2026