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.
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.
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.
Provides unified model definitions and a single API for pretrained text, vision, audio, and multimodal models for both training and inference. Emphasizes cross-framework compatibility (PyTorch/TF/JAX), pipeline-based inference, and direct access to 1M+ Hub checkpoints.
Turns model definitions into a shared layer across training and inference stacks, covering text, vision, audio, video, and multimodal models. Pipelines, Trainer, and generation APIs make pretrained models usable without locking teams to one framework.