Pytorch skills

PhysicsNeMo ShardTensor

nvidia

Optimize domain parallelism for PyTorch models.

NeMo AutoModel Distributed Training

nvidia

Streamline your distributed training setup with NeMo AutoModel.

PyTorch Geometric

k-dense-ai

Powerful library for graph neural networks in PyTorch.

PyTorch Lightning

k-dense-ai

Streamline your deep learning workflows with PyTorch.

Senior Computer Vision Engineer

davila7

Build and optimize production-grade vision AI systems.

Memory Tuning

nvidia

Optimize GPU memory usage during training runs.

Senior Computer Vision Engineer

alirezarezvani

Streamline your visual AI system development.

TAO Image Classification

nvidia

Streamline image classification with PyTorch-based models.

J

Jetson Package

nvidia

Streamline Jetson container and package selection.

Multi-Node Slurm

nvidia

Easily convert single-node scripts to multi-node Slurm jobs.

Stable Baselines3

k-dense-ai

Implement and train reinforcement learning agents with ease.

TorchDrug

k-dense-ai

Modular PyTorch stack for molecular graph learning.

HuggingFace Accelerate

davila7

Simplify distributed training with minimal code changes.

TorchTitan

nousresearch

Efficiently pretrain LLMs with PyTorch's 4D parallelism.

S

Spark Environment Setup

wshobson

Easily configure ML environments on NVIDIA DGX Spark.

TensorBoard

davila7

Visualize and debug machine learning models effectively.

Torchforge RL Training

davila7

Streamline your PyTorch-native reinforcement learning workflows.

PyTorch Lightning

nousresearch

Streamline your PyTorch training with organized code.

Flash Attention

nousresearch

Accelerate transformer training and inference significantly.

HuggingFace Accelerate

nousresearch

Simplify distributed PyTorch training with minimal code changes.

PyTorch Development Patterns

affaan-m

Best practices for building robust PyTorch applications.

AOTI Debugging

pytorch

Diagnose and resolve AOTInductor errors effectively.

PyTorch CI Metrics

pytorch

Access and analyze PyTorch CI and infrastructure metrics.

Fix PyTorch GitHub Issue

pytorch

Efficiently resolve bugs in PyTorch GitHub issues.