Skill UI

AudioCraft is Meta’s PyTorch toolkit for text-to-music and text-to-sound creation via MusicGen, AudioGen and EnCodec, making it ideal for prototyping music apps, effects, and controllable stereo pieces.

AudioCraft is Meta’s PyTorch toolkit for text-to-music and text-to-sound creation via MusicGen, AudioGen and EnCodec, making it ideal for prototyping music apps, effects, and controllable stereo pieces.

Orchestrates PyTorch-native TorchTitan 4D parallelism (FSDP2, TP, PP, CP) for large-scale LLM pretraining, scaling from 8 to 512+ GPUs with Float8, torch.compile, and distributed checkpointing to boot Llama 3.1, DeepSeek V3, or custom models.

Gradient Clipping Helper automates ML training work by guiding gradient clipping workflows, generating production-ready code and configs, and validating outputs for PyTorch, TensorFlow, and related pipelines.

Provides automated guidance for configuring learning rate schedulers during ML training, covering best practices, setup patterns, and framework-ready code snippets for PyTorch, TensorFlow, and Scikit-learn while validating against common standards.

A comprehensive skill set for building, deploying, and managing production-grade Machine Learning systems. Expertise spans modern frameworks (PyTorch, TensorFlow), model serving using Kubernetes, advanced feature engineering (Feast, Tecton), and robust MLOps practices (monitoring, A/B testing, CI/CD). Focuses on delivering scalable, reliable, and business-value-driven AI solutions.

Guides researchers through nnsight’s tracing and activation APIs so you can inspect or modify PyTorch internals locally or via NDIF-powered remote runs on 70B+ foundation models without local GPUs.

PyTorch Lightning is a high-level framework built on PyTorch that significantly reduces boilerplate code in deep learning training loops. It automatically handles device management, distributed training (DDP, FSDP, DeepSpeed), mixed precision, and advanced callbacks. It allows researchers to scale models from local machines to supercomputers with minimal code changes, enabling clean, scalable, and best-practice deep learning implementations.

Automates PyTorch training workflows by guiding data preparation, model configuration, hyperparameter tuning, and experiment tracking while following industry best practices for production readiness.

Pyvene is a declarative framework designed for performing causal interventions on PyTorch neural networks. It allows researchers to conduct advanced experiments such as activation patching, causal tracing (ROME-style), and interchange intervention training (IIT). Use this library when you need to test causal hypotheses about model behavior, deeply interpret model components, or ensure reproducibility in advanced AI research.

Comprehensive skill set covering the full lifecycle of advanced visual AI systems. Expertise includes object detection (YOLO, Faster R-CNN), instance and semantic segmentation (Mask R-CNN, SAM), model training using PyTorch, and production deployment optimization using ONNX and TensorRT for edge or cloud targets.

TensorBoard delivers a unified interface to track training metrics, debug network behavior, compare experiments, inspect model graphs, and profile performance for PyTorch or TensorFlow workflows.