What problem does it solve?
This Skill provides a comprehensive set of guidelines and recipes for optimizing PyTorch training processes, covering a wide range of topics from architecture selection to debugging.
Core Features & Use Cases
- Training Loop Best Practices: Offers best practices for efficient training loops, including gradient accumulation, memory management, and performance monitoring.
- Optimizer Configuration: Includes detailed instructions for configuring modern optimizers like AdamW and Muon for various parameter types.
- Learning Rate Scheduling: Provides guidelines for choosing and implementing learning rate schedules for optimal training dynamics.
- Mixed Precision & Compilation: Discusses the use of mixed precision training and the benefits of model compilation for performance.
- Memory & Performance: Offers strategies for managing memory and optimizing performance, including model FLOPs estimation and OOM solutions.
- Hyperparameter Search: Details a systematic approach to hyperparameter tuning, including a priority order and best practices.
- Debugging Checklist: Provides a comprehensive checklist for troubleshooting common issues in PyTorch training.
- Experiment Management: Offers a framework for tracking and managing experiments using TSV files and best practices for systematic experimentation.
- Evaluation Metrics: Defines key evaluation metrics for different types of machine learning tasks.
- Biomedical & Pharmaceutical ML: Provides insights into using machine learning in the biomedical and pharmaceutical domains, including molecular property prediction, drug discovery, and medical imaging.
- Genomic & Sequence Models: Offers guidance on using machine learning for genomic and sequence analysis.
- Single-Cell Omics: Discusses the use of machine learning in single-cell omics analysis.
- Clinical NLP: Provides insights into using machine learning for clinical NLP tasks like named entity recognition and relation extraction.
- EHR & Survival Analysis: Offers guidance on using machine learning for analyzing electronic health records and survival analysis.
- Biomedical Training Tricks: Provides strategies for training machine learning models on small datasets and handling class imbalance.
Quick Start
To get started with the ml-training-recipes Skill, use the following command:
npx skills add TTAWDTT/elegant-researcher-skill -g -y --copy