Results for “pytorch-lightning”
6 skillsMore results
Torch Geometric
Build and train graph neural networks with PyTorch Geometric, covering node/link/graph classification, message passing layers, heterogeneous graphs, and custom datasets.
30.2k · bundle
Ml Training Recipes
Provides battle-tested PyTorch training recipes for LLMs, vision, diffusion, and biomedical domains, covering training loops, optimizer selection, LR scheduling, mixed precision, and debugging.
10.4k · bundle
Transformers
Load pre-trained models from Hugging Face Hub, run pipeline inference, generate text, and fine-tune models on NLP, vision, audio, and multimodal tasks using the Transformers library.
30.2k · bundle
Torchdrug
Build and train graph neural networks for drug discovery, protein modeling, and molecular science using PyTorch-native tools.
30.2k · bundle
Torchprofiler Trace Analysis
Analyze PyTorch profiler and vLLM trace files with Perfetto SQL evidence. Use when working with torch profiler traces, merged_trace files, vLLM profiling logs, Perfetto trace analysis, GPU kernel hotspots, synchronization waits, rank imbalance, prefill/decode bottlenecks, or optimization reports.
0 · bundle