-
-
- baal +
- Clear ML
@@ -379,19 +379,19 @@
baal (bayesian active learning) aims to implement active learning using metrics of uncertainty derived from approximations of bayesian posteriors in neural networks.
+ClearML is a full system ML / DL experiment manager, versioning and ML-Ops solution.
-
-
- PyTorch Lightning +
- Flair
-
@@ -400,19 +400,19 @@
PyTorch Lightning is a Keras-like ML library for PyTorch. It leaves core training and validation logic to you and automates the rest.
+Flair is a very simple framework for state-of-the-art natural language processing (NLP).
-
-
- Substra +
- GaNDLF
-
@@ -421,19 +421,19 @@
Substra is a federated learning Python library to run federated learning experiments at scale on real distributed data.
+A generalizable application framework for segmentation, regression, and classification using PyTorch
-
-
- TorchMetrics +
- NeMo
-
@@ -442,19 +442,19 @@
Machine learning metrics for distributed, scalable PyTorch applications.
+NeMo: a toolkit for conversational AI.
-
-
- PyTorch-Points3d +
- Pyro
-
@@ -463,19 +463,19 @@
A PyTorch framework for deep learning on point clouds.
+Pyro is a universal probabilistic programming language (PPL) written in Python and supported by PyTorch on the backend.
-
-
- composer +
- TorchDrift
-
@@ -484,19 +484,19 @@
library of algorithms to speed up neural network training
+TorchDrift is a data and concept drift library for PyTorch. It lets you monitor your PyTorch models to see if they operate within spec.
-
-
- TorchDrift +
- Optuna
-
@@ -505,19 +505,19 @@
TorchDrift is a data and concept drift library for PyTorch. It lets you monitor your PyTorch models to see if they operate within spec.
+An open source hyperparameter optimization framework to automate hyperparameter search.
-
-
- PySyft +
- AdaptDL
-
@@ -526,19 +526,19 @@
PySyft is a Python library for encrypted, privacy preserving deep learning.
+AdaptDL is a resource-adaptive deep learning training and scheduling framework.
-
-
- TensorLy +
- DGL
-
@@ -547,19 +547,19 @@
TensorLy is a high level API for tensor methods and deep tensorized neural networks in Python that aims to make tensor learning simple.
+Deep Graph Library (DGL) is a Python package built for easy implementation of graph neural network model family, on top of PyTorch and other frameworks.
-
-
- Polyaxon +
- joeynmt
-
@@ -568,19 +568,19 @@
Polyaxon is a platform for building, training, and monitoring large-scale deep learning applications.
+Minimalist Neural Machine Translation toolkit for educational purposes
-
-
- fastai +
- Substra
-
@@ -589,19 +589,19 @@
fastai is a library that simplifies training fast and accurate neural nets using modern best practices.
+Substra is a federated learning Python library to run federated learning experiments at scale on real distributed data.
-
-
- TorchQuantum +
- skorch
-
@@ -610,19 +610,19 @@
TorchQuantum is a quantum classical simulation framework based on PyTorch. It supports statevector, density matrix simulation and pulse simulation on different hardware platforms such as CPUs and GPUs.
+skorch is a high-level library for PyTorch that provides full scikit-learn compatibility.
-
-
- accelerate +
- TorchIO
-
@@ -631,19 +631,19 @@
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
+TorchIO is a set of tools to efficiently read, preprocess, sample, augment, and write 3D medical images in deep learning applications written in PyTorch.
-
-
- Ensemble-Pytorch +
- TextBrewer
-
@@ -652,19 +652,19 @@
A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
+A PyTorch-based knowledge distillation toolkit for natural language processing
-
-
- PyTorch Geometric Temporal +
- torchdrug
-
@@ -673,19 +673,19 @@
PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric.
+A powerful and flexible machine learning platform for drug discovery.
-
-
- Pyro +
- Catalyst
-
@@ -694,19 +694,19 @@
Pyro is a universal probabilistic programming language (PPL) written in Python and supported by PyTorch on the backend.
+Catalyst helps you write compact, but full-featured deep learning and reinforcement learning pipelines with a few lines of code.
-
-
- Hydra +
- baal
-
@@ -715,19 +715,19 @@
A framework for elegantly configuring complex applications.
+baal (bayesian active learning) aims to implement active learning using metrics of uncertainty derived from approximations of bayesian posteriors in neural networks.
-
-
- einops +
- AllenNLP
-
@@ -736,19 +736,19 @@
Flexible and powerful tensor operations for readable and reliable code.
+AllenNLP is an open-source research library built on PyTorch for designing and evaluating deep learning models for NLP.
-
-
- AdverTorch +
- PyKale
-
@@ -757,19 +757,19 @@
A toolbox for adversarial robustness research. It contains modules for generating adversarial examples and defending against attacks.
+PyKale is a PyTorch library for multimodal learning and transfer learning with deep learning and dimensionality reduction on graphs, images, texts, and videos.
-
-
- TextBrewer +
- einops
-
@@ -778,19 +778,19 @@
A PyTorch-based knowledge distillation toolkit for natural language processing
+Flexible and powerful tensor operations for readable and reliable code.
-
-
- PyTorch Metric Learning +
- TorchQuantum
-
@@ -799,19 +799,19 @@
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible.
+TorchQuantum is a quantum classical simulation framework based on PyTorch. It supports statevector, density matrix simulation and pulse simulation on different hardware platforms such as CPUs and GPUs.
-
-
- ludwig +
- avalanche
-
@@ -820,19 +820,19 @@
Data-centric declarative deep learning framework
+Avalanche: an End-to-End Library for Continual Learning
-
-
- PyTorch Geometric +
- ClinicaDL
-
@@ -841,19 +841,19 @@
PyTorch Geometric is a library for deep learning on irregular input data such as graphs, point clouds, and manifolds.
+Framework for reproducible classification of Alzheimer's Disease
-
-
- TIAToolbox +
- depyf
-
@@ -862,19 +862,19 @@
TIAToolbox provides an easy-to-use API where researchers can use, adapt and create models for CPath.
+depyf is a tool to help users understand and adapt to PyTorch compiler torch.compile.
-
-
- avalanche +
- FuseMedML
-
@@ -883,19 +883,19 @@
Avalanche: an End-to-End Library for Continual Learning
+FuseMedML is a python framework accelerating ML based discovery in the medical field by encouraging code reuse
-
-
- ClinicaDL +
- forte
-
@@ -904,19 +904,19 @@
Framework for reproducible classification of Alzheimer's Disease
+Forte is a toolkit for building NLP pipelines featuring composable components, convenient data interfaces, and cross-task interaction.
-
-
- AdaptDL +
- Hummingbird
-
@@ -925,19 +925,19 @@
AdaptDL is a resource-adaptive deep learning training and scheduling framework.
+Hummingbird compiles trained ML models into tensor computation for faster inference.
-
-
- AllenNLP +
- Opacus
-
@@ -946,19 +946,19 @@
AllenNLP is an open-source research library built on PyTorch for designing and evaluating deep learning models for NLP.
+Train PyTorch models with Differential Privacy
-
-
- USB +
- ludwig
-
@@ -967,19 +967,19 @@
USB is a Pytorch-based Python package for Semi-Supervised Learning (SSL). It is easy-to-use/extend, affordable to small groups, and comprehensive for developing and evaluating SSL algorithms.
+Data-centric declarative deep learning framework
-
-
- Opacus +
- Ignite
-
@@ -988,19 +988,19 @@
Train PyTorch models with Differential Privacy
+Ignite is a high-level library for training neural networks in PyTorch. It helps with writing compact, but full-featured training loops.
-
-
- GPyTorch +
- PyTorch Lightning
-
@@ -1009,19 +1009,19 @@
GPyTorch is a Gaussian process library implemented using PyTorch, designed for creating scalable, flexible Gaussian process models.
+PyTorch Lightning is a Keras-like ML library for PyTorch. It leaves core training and validation logic to you and automates the rest.
-
-
- PyKale +
- Ray
-
@@ -1030,19 +1030,19 @@
PyKale is a PyTorch library for multimodal learning and transfer learning with deep learning and dimensionality reduction on graphs, images, texts, and videos.
+Ray is a fast and simple framework for building and running distributed applications.
-
-
- Hummingbird +
- fastai
-
@@ -1051,19 +1051,19 @@
Hummingbird compiles trained ML models into tensor computation for faster inference.
+fastai is a library that simplifies training fast and accurate neural nets using modern best practices.
-
-
- padl +
- PyTorch Geometric
-
@@ -1072,19 +1072,19 @@
Pipeline Abstractions for Deep Learning in PyTorch
+PyTorch Geometric is a library for deep learning on irregular input data such as graphs, point clouds, and manifolds.
-
-
- SimulAI +
- Polyaxon
-
@@ -1093,19 +1093,19 @@
SimulAI is basically a toolkit with pipelines for physics-informed machine learning.
+Polyaxon is a platform for building, training, and monitoring large-scale deep learning applications.
-
-
- PyTorch-NLP +
- higher
-
@@ -1114,19 +1114,19 @@
Basic Utilities for PyTorch Natural Language Processing (NLP).
+higher is a library which facilitates the implementation of arbitrarily complex gradient-based meta-learning algorithms and nested optimisation loops with near-vanilla PyTorch.
-
-
- torchdistill +
- Kornia
-
@@ -1135,19 +1135,19 @@
torchdistill is a coding-free framework built on PyTorch for reproducible deep learning and knowledge distillation studies.
+Kornia is a differentiable computer vision library that consists of a set of routines and differentiable modules to solve generic CV problems.
-
-
- Flair +
- Renate
-
@@ -1156,19 +1156,19 @@
Flair is a very simple framework for state-of-the-art natural language processing (NLP).
+Renate is a library providing tools for re-training pytorch models over time as new data becomes available.
-
-
- Detectron2 +
- SimulAI
-
@@ -1177,19 +1177,19 @@
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
+SimulAI is basically a toolkit with pipelines for physics-informed machine learning.
-
-
- MONAI +
- Colossal-LLaMA-2
-
@@ -1198,19 +1198,19 @@
MONAI provides domain-optimized foundational capabilities for developing healthcare imaging training workflows.
+A complete and open-sourced solution for injecting domain-specific knowledge into pre-trained LLM.
-
-
- PyPOTS +
- CrypTen
-
@@ -1219,19 +1219,19 @@
A Python toolbox for data mining on Partially-Observed Time Series (POTS) and helps engineers focus more on the core problems in rather than missing parts in their data.
+CrypTen is a framework for Privacy Preserving ML. Its goal is to make secure computing techniques accessible to ML practitioners.
-
-
- PennyLane +
- Stable Baselines3
-
@@ -1240,19 +1240,19 @@
PennyLane is a library for quantum ML, automatic differentiation, and optimization of hybrid quantum-classical computations.
+Stable Baselines3 (SB3) is a set of reliable implementations of reinforcement learning algorithms in PyTorch.
-
-
- ivy +
- PyTorch3D
-
@@ -1261,19 +1261,19 @@
The Unified Machine Learning Framework
+PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch.
-
-
- ColossalAI +
- PyTorch Geometric Temporal
-
@@ -1282,19 +1282,19 @@
Colossal-AI is a Unified Deep Learning System for Big Model Era
+PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric.
-
-
- PyPose +
- OctoML Profile
-
@@ -1303,19 +1303,19 @@
PyPose is a robotics-oriented, PyTorch-based library that combines deep perceptual models with physics-based optimization techniques, so that users can focus on their novel applications.
+octoml-profile is a python library and cloud service designed to provide a simple experience for assessing and optimizing the performance of PyTorch models.
-
-
- Flower +
- Torch-TensorRT
-
@@ -1324,19 +1324,19 @@
Flower - A Friendly Federated Learning Framework
+PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
-
-
- pytorchfi +
- PennyLane
-
@@ -1345,19 +1345,19 @@
A runtime fault injection tool for PyTorch.
+PennyLane is a library for quantum ML, automatic differentiation, and optimization of hybrid quantum-classical computations.
-
-
- L5Kit +
- VISSL
-
@@ -1366,19 +1366,19 @@
ML Prediction, Planning and Simulation for Self-Driving built on PyTorch.
+A library for state-of-the-art self-supervised learning
-
-
- Ray +
- ParlAI
-
@@ -1387,19 +1387,19 @@
Ray is a fast and simple framework for building and running distributed applications.
+ParlAI is a unified platform for sharing, training, and evaluating dialog models across many tasks.
-
-
- Horovod +
- pytorchfi
-
@@ -1408,19 +1408,19 @@
Horovod is a distributed training library for deep learning frameworks. Horovod aims to make distributed DL fast and easy to use.
+A runtime fault injection tool for PyTorch.
-
-
- Colossal-LLaMA-2 +
- MONAI
-
@@ -1429,19 +1429,19 @@
A complete and open-sourced solution for injecting domain-specific knowledge into pre-trained LLM.
+MONAI provides domain-optimized foundational capabilities for developing healthcare imaging training workflows.
-
-
- Renate +
- Diffusers
-
@@ -1450,19 +1450,19 @@
Renate is a library providing tools for re-training pytorch models over time as new data becomes available.
+Diffusers provides pretrained diffusion models across multiple modalities, such as vision and audio, and serves as a modular toolbox for inference and training of diffusion models.
-
-
- Catalyst +
- L5Kit
-
@@ -1471,19 +1471,19 @@
Catalyst helps you write compact, but full-featured deep learning and reinforcement learning pipelines with a few lines of code.
+ML Prediction, Planning and Simulation for Self-Driving built on PyTorch.
-
-
- RoMa +
- OpenMMLab
-
@@ -1492,19 +1492,19 @@
RoMa is a standalone library to handle rotation representations with PyTorch (rotation matrices, quaternions, rotation vectors, etc). It aims for robustness, ease-of-use, and efficiency.
+OpenMMLab covers a wide range of computer vision research topics including classification, detection, segmentation, and super-resolution.
-
-
- joeynmt +
- Ensemble-Pytorch
-
@@ -1513,19 +1513,19 @@
Minimalist Neural Machine Translation toolkit for educational purposes
+A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
-
-
- skorch +
- TIAToolbox
-
@@ -1534,19 +1534,19 @@
skorch is a high-level library for PyTorch that provides full scikit-learn compatibility.
+TIAToolbox provides an easy-to-use API where researchers can use, adapt and create models for CPath.
-
-
- TorchIO +
- MMF
-
@@ -1555,19 +1555,19 @@
TorchIO is a set of tools to efficiently read, preprocess, sample, augment, and write 3D medical images in deep learning applications written in PyTorch.
+A modular framework for vision & language multimodal research from Facebook AI Research (FAIR).
-
-
- GaNDLF +
- composer
-
@@ -1576,19 +1576,19 @@
A generalizable application framework for segmentation, regression, and classification using PyTorch
+library of algorithms to speed up neural network training
-
-
- Determined +
- PyTorch-Points3d
-
@@ -1597,19 +1597,19 @@
Determined is a platform that helps deep learning teams train models more quickly, easily share GPU resources, and effectively collaborate.
+A PyTorch framework for deep learning on point clouds.
-
-
- pystiche +
- PyPose
-
@@ -1618,19 +1618,19 @@
pystiche is a framework for Neural Style Transfer (NST) built upon PyTorch.
+PyPose is a robotics-oriented, PyTorch-based library that combines deep perceptual models with physics-based optimization techniques, so that users can focus on their novel applications.
-
-
- NeMo +
- Determined
-
@@ -1639,19 +1639,19 @@
NeMo: a toolkit for conversational AI.
+Determined is a platform that helps deep learning teams train models more quickly, easily share GPU resources, and effectively collaborate.
-
-
- PyTorch3D +
- intel-extension-for-pytorch
-
@@ -1660,19 +1660,19 @@
PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch.
+A Python package for improving PyTorch performance on Intel platforms
-
-
- Stable Baselines3 +
- BoTorch
-
@@ -1681,19 +1681,19 @@
Stable Baselines3 (SB3) is a set of reliable implementations of reinforcement learning algorithms in PyTorch.
+BoTorch is a library for Bayesian Optimization. It provides a modular, extensible interface for composing Bayesian optimization primitives.
-
-
- Albumentations +
- TensorLy
-
@@ -1702,19 +1702,19 @@
Fast and extensible image augmentation library for different CV tasks like classification, segmentation, object detection and pose estimation.
+TensorLy is a high level API for tensor methods and deep tensorized neural networks in Python that aims to make tensor learning simple.
-
-
- ParlAI +
- accelerate
-
@@ -1723,19 +1723,19 @@
ParlAI is a unified platform for sharing, training, and evaluating dialog models across many tasks.
+🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
-
-
- OpenMMLab +
- PyPOTS
-
@@ -1744,19 +1744,19 @@
OpenMMLab covers a wide range of computer vision research topics including classification, detection, segmentation, and super-resolution.
+A Python toolbox for data mining on Partially-Observed Time Series (POTS) and helps engineers focus more on the core problems in rather than missing parts in their data.
-
-
- DeepSpeed +
- PyTorchVideo
-
@@ -1765,19 +1765,19 @@
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
+A deep learning library for video understanding research. Hosts various video-focused models, datasets, training pipelines and more.
-
-
- Transformers +
- padl
-
@@ -1786,19 +1786,19 @@
State-of-the-art Natural Language Processing for PyTorch.
+Pipeline Abstractions for Deep Learning in PyTorch
-
-
- DGL +
- RoMa
-
@@ -1828,19 +1828,19 @@
Deep Graph Library (DGL) is a Python package built for easy implementation of graph neural network model family, on top of PyTorch and other frameworks.
+RoMa is a standalone library to handle rotation representations with PyTorch (rotation matrices, quaternions, rotation vectors, etc). It aims for robustness, ease-of-use, and efficiency.
-
-
- FairScale +
- ONNX Runtime
-
@@ -1849,19 +1849,19 @@
FairScale is a PyTorch extension library for high performance and large scale training on one or multiple machines/nodes.
+ONNX Runtime is a cross-platform inferencing and training accelerator.
-
-
- Lightly +
- Captum
-
@@ -1870,19 +1870,19 @@
Lightly is a computer vision framework for self-supervised learning.
+Captum (“comprehension” in Latin) is an open source, extensible library for model interpretability built on PyTorch.
-
-
- ONNX Runtime +
- Detectron2
-
@@ -1891,19 +1891,19 @@
ONNX Runtime is a cross-platform inferencing and training accelerator.
+Detectron2 is FAIR's next-generation platform for object detection and segmentation.
-
-
- Diffusers +
- PopTorch
-
@@ -1912,19 +1912,19 @@
Diffusers provides pretrained diffusion models across multiple modalities, such as vision and audio, and serves as a modular toolbox for inference and training of diffusion models.
+The PopTorch interface library is a simple wrapper for running PyTorch programs directly on Graphcore IPUs.
-
-
- stoke +
- DeepSpeed
-
@@ -1933,19 +1933,19 @@
A lightweight declarative PyTorch wrapper for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
+DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
-
-
- PopTorch +
- Hydra
-
@@ -1954,19 +1954,19 @@
The PopTorch interface library is a simple wrapper for running PyTorch programs directly on Graphcore IPUs.
+A framework for elegantly configuring complex applications.
-
-
- PyTorchVideo +
- AdverTorch
-
@@ -1975,19 +1975,19 @@
A deep learning library for video understanding research. Hosts various video-focused models, datasets, training pipelines and more.
+A toolbox for adversarial robustness research. It contains modules for generating adversarial examples and defending against attacks.
-
-
- neural-compressor +
- Flower
-
@@ -1996,19 +1996,19 @@
Intel® Neural Compressor provides unified APIs for network compression technologies for faster inference
+Flower - A Friendly Federated Learning Framework
-
-
- intel-extension-for-pytorch +
- neural-compressor
-
@@ -2059,19 +2059,19 @@
A Python package for improving PyTorch performance on Intel platforms
+Intel® Neural Compressor provides unified APIs for network compression technologies for faster inference
-
-
- FuseMedML +
- torchgeo
-
@@ -2080,19 +2080,19 @@
FuseMedML is a python framework accelerating ML based discovery in the medical field by encouraging code reuse
+Datasets, transforms, and models for geospatial data
-
-
- raster-vision +
- GPyTorch
-
@@ -2122,19 +2122,19 @@
An open source framework for deep learning on satellite and aerial imagery.
+GPyTorch is a Gaussian process library implemented using PyTorch, designed for creating scalable, flexible Gaussian process models.
-
-
- BoTorch +
- USB
-
@@ -2143,19 +2143,19 @@
BoTorch is a library for Bayesian Optimization. It provides a modular, extensible interface for composing Bayesian optimization primitives.
+USB is a Pytorch-based Python package for Semi-Supervised Learning (SSL). It is easy-to-use/extend, affordable to small groups, and comprehensive for developing and evaluating SSL algorithms.
-
-
- Clear ML +
- TorchMetrics
-
@@ -2164,19 +2164,19 @@
ClearML is a full system ML / DL experiment manager, versioning and ML-Ops solution.
+Machine learning metrics for distributed, scalable PyTorch applications.
-
-
- OctoML Profile +
- PyTorch-NLP
-
@@ -2185,19 +2185,19 @@
octoml-profile is a python library and cloud service designed to provide a simple experience for assessing and optimizing the performance of PyTorch models.
+Basic Utilities for PyTorch Natural Language Processing (NLP).
-
-
- Kornia +
- Poutyne
-
@@ -2206,19 +2206,19 @@
Kornia is a differentiable computer vision library that consists of a set of routines and differentiable modules to solve generic CV problems.
+Poutyne is a Keras-like framework for PyTorch and handles much of the boilerplating code needed to train neural networks.
-
-
- Optuna +
- Transformers
-
@@ -2227,19 +2227,19 @@
An open source hyperparameter optimization framework to automate hyperparameter search.
+State-of-the-art Natural Language Processing for PyTorch.
-
-
- VISSL +
- PFRL
-
@@ -2248,19 +2248,19 @@
A library for state-of-the-art self-supervised learning
+PFRL is a deep reinforcement learning library that implements various state-of-the-art deep reinforcement algorithms in Python using PyTorch.
-
-
- torchgeo +
- ivy
-
@@ -2269,19 +2269,19 @@
Datasets, transforms, and models for geospatial data
+The Unified Machine Learning Framework
-
-
- forte +
- raster-vision
-
@@ -2290,19 +2290,19 @@
Forte is a toolkit for building NLP pipelines featuring composable components, convenient data interfaces, and cross-task interaction.
+An open source framework for deep learning on satellite and aerial imagery.
-
-
- Ignite +
- FairScale
-
@@ -2311,19 +2311,19 @@
Ignite is a high-level library for training neural networks in PyTorch. It helps with writing compact, but full-featured training loops.
+FairScale is a PyTorch extension library for high performance and large scale training on one or multiple machines/nodes.
-
-
- Torch-TensorRT +
- Lightly
-
@@ -2332,19 +2332,19 @@
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
+Lightly is a computer vision framework for self-supervised learning.
-
-
- depyf +
- Horovod
-
@@ -2353,19 +2353,19 @@
depyf is a tool to help users understand and adapt to PyTorch compiler torch.compile.
+Horovod is a distributed training library for deep learning frameworks. Horovod aims to make distributed DL fast and easy to use.
-
-
- MMF +
- pomegranate
-
@@ -2374,19 +2374,19 @@
A modular framework for vision & language multimodal research from Facebook AI Research (FAIR).
+pomegranate is a library of probabilistic models that is built in a modular manner and treats all models as the probability distributions that they are.
-
-
- PFRL +
- Albumentations
-
@@ -2395,19 +2395,19 @@
PFRL is a deep reinforcement learning library that implements various state-of-the-art deep reinforcement algorithms in Python using PyTorch.
+Fast and extensible image augmentation library for different CV tasks like classification, segmentation, object detection and pose estimation.
-
-
- Captum +
- PyTorch Metric Learning
-
@@ -2416,19 +2416,19 @@
Captum (“comprehension” in Latin) is an open source, extensible library for model interpretability built on PyTorch.
+The easiest way to use deep metric learning in your application. Modular, flexible, and extensible.
-
-
- pomegranate +
- torchdistill
-
@@ -2437,19 +2437,19 @@
pomegranate is a library of probabilistic models that is built in a modular manner and treats all models as the probability distributions that they are.
+torchdistill is a coding-free framework built on PyTorch for reproducible deep learning and knowledge distillation studies.
-
-
- torchdrug +
- ColossalAI
-
@@ -2458,19 +2458,19 @@
A powerful and flexible machine learning platform for drug discovery.
+Colossal-AI is a Unified Deep Learning System for Big Model Era
-
-
- CrypTen +
- pystiche
-
@@ -2479,7 +2479,7 @@
CrypTen is a framework for Privacy Preserving ML. Its goal is to make secure computing techniques accessible to ML practitioners.
+pystiche is a framework for Neural Style Transfer (NST) built upon PyTorch.
", + "command": null + }, + "accnone": { + "note": null, + "command": "conda install pytorch torchvision torchaudio cpuonly -c pytorch" + } + }, + "libtorch": { + "accnone": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.4.1%2Bcpu.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.4.1%2Bcpu.zip" + } + }, + "cuda.x": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.4.1%2Bcu118.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.4.1%2Bcu118.zip" + } + }, + "cuda.y": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.4.1%2Bcu121.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.4.1%2Bcu121.zip" + } + }, + "cuda.z": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu124/libtorch-shared-with-deps-2.4.1%2Bcu124.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-2.4.1%2Bcu124.zip" + } + }, + "rocm5.x": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/rocm6.1/libtorch-shared-with-deps-2.4.1%2Brocm6.1.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/rocm6.1/libtorch-cxx11-abi-shared-with-deps-2.4.1%2Brocm6.1.zip" + } + } + } + }, + "macos": { + "pip": { + "cuda.x": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "pip3 install torch torchvision torchaudio", + "default": true + }, + "cuda.y": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "pip3 install torch torchvision torchaudio", + "default": true + }, + "cuda.z": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "pip3 install torch torchvision torchaudio", + "default": true + }, + "rocm5.x": { + "note": "# ROCm is not available on MacOS, please use default package", + "command": "pip3 install torch torchvision torchaudio", + "default": true + }, + "accnone": { + "note": null, + "command": "pip3 install torch torchvision torchaudio" + } + }, + "conda": { + "cuda.x": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", + "default": true + }, + "cuda.y": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", + "default": true + }, + "cuda.z": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", + "default": true + }, + "rocm5.x": { + "note": "# ROCm is not available on MacOS, please use default package", + "command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", + "default": true + }, + "accnone": { + "note": null, + "command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch" + } + }, + "libtorch": { + "accnone": { + "note": null, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.1.zip" + } + }, + "cuda.x": { + "note": null, + "default": true, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.1.zip" + } + }, + "cuda.y": { + "note": null, + "default": true, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.1.zip" + } + }, + "cuda.z": { + "note": null, + "default": true, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.1.zip" + } + }, + "rocm5.x": { + "note": null, + "default": true, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.1.zip" + } + } + } + }, + "windows": { + "pip": { + "accnone": { + "note": null, + "command": "pip3 install torch torchvision torchaudio" + }, + "cuda.x": { + "note": null, + "command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118" + }, + "cuda.y": { + "note": null, + "command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121" + }, + "cuda.z": { + "note": null, + "command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124" + }, + "rocm5.x": { + "note": "NOTE: ROCm is not available on Windows", + "command": null + } + }, + "conda": { + "cuda.x": { + "note": null, + "command": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia" + }, + "cuda.y": { + "note": null, + "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia" + }, + "cuda.z": { + "note": null, + "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia" + }, + "rocm5.x": { + "note": "NOTE: ROCm is not available on Windows", + "command": null + }, + "accnone": { + "note": null, + "command": "conda install pytorch torchvision torchaudio cpuonly -c pytorch" + } + }, + "libtorch": { + "accnone": { + "note": null, + "versions": { + "Download here (Release version):": "https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-2.4.1%2Bcpu.zip", + "Download here (Debug version):": "https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-2.4.1%2Bcpu.zip" + } + }, + "cuda.x": { + "note": null, + "versions": { + "Download here (Release version):": "https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-2.4.1%2Bcu118.zip", + "Download here (Debug version):": "https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-debug-2.4.1%2Bcu118.zip" + } + }, + "cuda.y": { + "note": null, + "versions": { + "Download here (Release version):": "https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-2.4.1%2Bcu121.zip", + "Download here (Debug version):": "https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-debug-2.4.1%2Bcu121.zip" + } + }, + "cuda.z": { + "note": null, + "versions": { + "Download here (Release version):": "https://download.pytorch.org/libtorch/cu124/libtorch-win-shared-with-deps-2.4.1%2Bcu124.zip", + "Download here (Debug version):": "https://download.pytorch.org/libtorch/cu124/libtorch-win-shared-with-deps-debug-2.4.1%2Bcu124.zip" + } + }, + "rocm5.x": { + "note": "NOTE: ROCm is not available on Windows", + "versions": null + } + } + } } } } \ No newline at end of file