Lets researchers and engineers build neural networks as regular Python programs, with GPU-backed tensors, autograd, distributed training, and production paths through TorchScript and related tooling.
Tracks every ML run — hyperparameters, metrics, checkpoints, dataset versions — into one dashboard you share as a live report, with Sweeps for tuning and a model registry. Weave extends it to LLM apps: tracing, evals, and production monitoring.
Sequence modeling toolkit for training custom models for translation, summarization, and language modeling. Reference implementation behind RoBERTa, BART, mBART, XLM-R, and wav2vec 2.0, with multi-GPU and mixed-precision training.
Defines a portable model format and operator set for moving trained machine learning models across frameworks, runtimes, and hardware targets without locking the model to one toolchain.
Provides reusable PyTorch modules and utilities to build, train and scale Graph Neural Networks — includes many implemented GNN layers, benchmark datasets, minibatch/sampling loaders, and support for large-scale, heterogeneous, temporal and point-cloud graphs.
Scales any Python or ML workload across CPUs and GPUs with a few decorators, instead of rewriting code for Spark or MPI. Bundles libraries for distributed training, hyperparameter tuning, RL, batch inference, and online model serving on one cluster.
Swaps faces in images and videos using deep learning, offering tools to extract faces, train generative models, and convert media via CLI or GUI for research, VFX, and ethical experimentation.
Provides 150+ executed Jupyter notebooks and code that reproduce the book 'Machine Learning for Algorithmic Trading (2nd ed.)' — covers feature engineering, alternative-data signal extraction, backtesting, NLP, deep learning and reinforcement learning for trading; best for quant researchers and practitioners.
Brings classic computer vision into PyTorch as differentiable, GPU-accelerated tensor operators — filters, geometric transforms, feature matching, camera calibration — so each step lives inside autograd and trains end-to-end with neural networks.
Serves machine learning and deep learning models for cloud, data center, edge and embedded environments. Supports multiple frameworks and backends, dynamic and sequence batching, HTTP/gRPC APIs, Docker deployment and NVIDIA-optimized runtimes.
Notebook-first deep learning textbook that teaches concepts through runnable multi-framework code, math, and exercises. Includes lecture-ready notebooks, community contributions, and broad university adoption—designed for hands-on learners and instructors.
Converts, quantizes, and runs deep learning models from PyTorch, TensorFlow, ONNX, and PaddlePaddle across Intel CPUs, GPUs, and NPUs without the training framework. Adds a GenAI pipeline for LLMs plus Hugging Face, vLLM, and LangChain integrations.