landscape.yml

landscape:
  - category:
    name: Modeling
    subcategories:
      - subcategory:
        name: Computer Vision
        items:
          - item:
            name: Albumentations
            description: a Python library for image augmentation used in deep learning and computer vision tasks to increase the quality of trained models. The purpose of image augmentation is to create new training samples from the existing data.
            homepage_url: https://albumentations.ai/docs/
            logo: albumentations.svg
            repo_url: https://github.com/albumentations-team/albumentations
            crunchbase: https://www.crunchbase.com/organization/albumentations
          - item:
            name: Anomalib
            description: An anomaly detection library comprising state-of-the-art algorithms and features such as experiment management, hyper-parameter optimization, and edge inference.
            homepage_url: https://anomalib.readthedocs.io/en/latest/
            logo: anomalib-wide.svg
            repo_url: https://github.com/openvinotoolkit/anomalib
            crunchbase: https://www.crunchbase.com/organization/intel
          - item:
            name: Detectron2
            description: is a platform for object detection, segmentation and other visual recognition tasks.
            homepage_url: https://detectron2.readthedocs.io/en/latest/
            logo: detectron2.svg
            repo_url: https://github.com/facebookresearch/detectron2
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Kornia
            description: Kornia is a geometric computer vision Library for spatial AI library that allows classical computer vision to be integrated into deep learning models.
            homepage_url: https://kornia.github.io/
            logo: kornia_banner_pixie.svg
            repo_url: https://github.com/kornia/kornia
          - item:
            name: Lightly
            description: LightlySSL is a computer vision framework for self-supervised learning.
            homepage_url: https://docs.lightly.ai/self-supervised-learning/
            logo: lightly_SSL_logo_crop_white_text.svg
            repo_url: https://github.com/lightly-ai/lightly
          - item:
            name: MMEditing
            description: Image and Video Restoration, Editing and Generation Toolbox
            homepage_url: https://mmagic.readthedocs.io/en/latest/
            logo: mmediting.svg
            repo_url: https://github.com/open-mmlab/mmediting
          - item:
            name: OpenMMLab
            description: Open-source computer vision algorithms and models.
            homepage_url: https://openmmlab.com/
            logo: openmmlab.svg
            repo_url: https://github.com/open-mmlab
          - item:
            name: pystiche
            description: Framework for Neural Style Transfer (NST) built upon PyTorch
            homepage_url: https://github.com/pystiche/pystiche
            logo: pystiche.svg
            repo_url: https://github.com/pystiche/pystiche
          - item:
            name: raster-vision
            description: Raster Vision is an open source library and framework for Python developers building computer vision models on satellite, aerial, and other large imagery sets (including oblique drone imagery). There is built-in support for chip classification, object detection, and semantic segmentation using PyTorch.
            homepage_url: https://docs.rastervision.io/
            logo: raster-vision.svg
            repo_url: https://github.com/azavea/raster-vision
          - item:
            name: torchvision
            description: The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
            homepage_url: https://pytorch.org/vision/stable/index.html
            logo: torchvision.svg
            repo_url: https://github.com/pytorch/vision
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: VISSL
            description: A library for state-of-the-art self-supervised learning from images
            homepage_url: https://vissl.ai/
            logo: vissl.svg
            repo_url: https://github.com/facebookresearch/vissl/graphs/commit-activity
            crunchbase: https://www.crunchbase.com/organization/facebook
      - subcategory:
        name: Language
        items:
          - item:
            name: AllenNLP
            description: An NLP research library, built on PyTorch, for developing state-of-the-art deep learning models on a wide variety of linguistic tasks.
            homepage_url: https://allenai.org/allennlp
            logo: AllenNLP.svg
            repo_url: https://github.com/allenai/allennlp
            crunchbase: https://www.crunchbase.com/organization/allen-institute-for-artificial-intelligence
          - item:
            name: Colossal-LLaMA-2
            description: This project is no longer available
            homepage_url: https://www.colossalai.org/
            logo: collasalllama.svg
            repo_url:  https://github.com/hpcaitech/ColossalAI/tree/main/applications/Colossal-LLaMA-2 
            crunchbase: https://www.crunchbase.com/organization/hpc-ai-technology
          - item:
            name: Flair
            description: Flair allows you to apply our state-of-the-art natural language processing (NLP) models to your text, such as named entity recognition (NER), sentiment analysis, part-of-speech tagging (PoS), special support for biomedical texts, sense disambiguation and classification, with support for a rapidly growing number of languages
            homepage_url: https://github.com/flairNLP/flair
            logo: flair.svg
            repo_url: https://github.com/flairNLP/flair
            crunchbase: https://www.crunchbase.com/organization/humboldt-university-of-berlin
          - item:
            name: joeynmt
            description: JoeyNMT is a minimalist neural machine translation toolkit for educational purposes.
            homepage_url: https://joeynmt.readthedocs.io/en/latest/#
            logo: joeynmt.svg
            repo_url: https://github.com/joeynmt
            crunchbase: https://www.crunchbase.com/organization/cornell-university
          - item:
            name: OpenCompass
            description: OpenCompass is an LLM evaluation platform, supporting a wide range of models (Llama3, Mistral, InternLM2,GPT-4,LLaMa2, Qwen,GLM, Claude, etc) over 100+ datasets.
            homepage_url: https://opencompass.org.cn/home
            logo: opencompass.svg
            repo_url: https://github.com/open-compass/opencompass
          - item:
            name: ParIAI
            description: A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
            homepage_url: https://parl.ai/
            logo: parlai.svg
            repo_url: https://github.com/facebookresearch/ParlAI
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: PyTorch-NLP
            description: Basic Utilities for PyTorch Natural Language Processing (NLP)
            homepage_url: https://pytorchnlp.readthedocs.io/en/latest/
            logo: PyTorch_NLP.svg
            repo_url: https://github.com/PetrochukM/PyTorch-NLP
          - item:
            name: TextBrewer
            description: A PyTorch-based knowledge distillation toolkit for natural language processing
            homepage_url: http://textbrewer.hfl-rc.com/
            logo: TextBrewer.svg
            repo_url: https://github.com/airaria/TextBrewer
          - item:
            name: torchtune
            description: A Native-PyTorch library for LLM fine-tuning.
            homepage_url: https://pytorch.org/torchtune/stable/index.html
            logo: torchtune.svg
            repo_url: https://github.com/pytorch/torchtune
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: torchtext
            description: The torchtext package consists of data processing utilities and popular datasets for natural language.
            homepage_url: https://pytorch.org/text/stable/index.html
            logo: torchtext.svg
            repo_url: https://github.com/pytorch/text
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Transformers
            description: Transformers provides thousands of pretrained models to perform tasks on different modalities such as text, vision, and audio.
            homepage_url: https://huggingface.co/transformers
            logo: Transformers.svg
            repo_url: https://github.com/huggingface/transformers
            crunchbase: https://www.crunchbase.com/organization/hugging-face
      - subcategory:
        name: Specialized
        items:
          - item:
            name: DGL
            description: Python package built to ease deep learning on graph, on top of existing DL frameworks.  Fast and memory-efficient message passing primitives for training Graph Neural Networks.
            homepage_url: https://www.dgl.ai/
            logo: DGL.svg
            repo_url: https://github.com/dmlc/dgl/
            crunchbase: https://www.crunchbase.com/organization/new-york-university
          - item:
            name: Diffusers 
            description: Pretrained diffusion models for generating images, audio, and even 3D structures of molecules. Whether you are looking for a simple inference solution or want to train your own diffusion model, Diffusers is a modular toolbox that supports both
            homepage_url: https://huggingface.co/docs/diffusers/index
            logo: Diffusers.svg
            repo_url: https://github.com/huggingface/diffusers
            crunchbase: https://www.crunchbase.com/organization/hugging-face
          - item:
            name: L5Kit
            description: A python library with functionality for the development and training of learned prediction, planning and simulation models for autonomous driving applications.
            homepage_url: https://woven-planet.github.io/l5kit/
            logo: logo_l5kit_wp.svg
            repo_url: https://github.com/woven-planet/l5kit
          - item:
            name: PyPose
            description: PyPose is a library for Robot Learning with Physics-based Optimization
            homepage_url: https://pypose.org/
            logo: PyPose.svg
            repo_url: https://github.com/pypose/pypose
          - item:
            name: PyPOTS
            description: a Python toolbox for machine learning on Partially-Observed Time Series
            homepage_url: https://pypots.com/
            logo: PyPOTS.svg
            repo_url: https://github.com/WenjieDu/PyPOTS
          - item:
            name: PyTorchVideo
            description: A deep learning library for video understanding research.
            homepage_url: https://pytorchvideo.org/
            logo: PyTorchVideo.svg
            repo_url: https://github.com/facebookresearch/pytorchvideo
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: SimulAI 
            description: A toolkit with data-driven pipelines for physics-informed machine learning.
            homepage_url: https://ibm.github.io/simulai/
            logo: SimulAI.svg
            repo_url: https://github.com/IBM/simulai
            crunchbase: https://www.crunchbase.com/organization/ibm
          - item:
            name: torchaudio
            description: Torchaudio is a library for audio and signal processing with PyTorch. It provides I/O, signal and data processing functions, datasets, model implementations and application components.
            homepage_url: https://pytorch.org/audio/stable/index.html
            logo: torchaudio.svg
            repo_url: https://github.com/pytorch/audio
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: torchrec
            description: TorchRec is a specialized library within the PyTorch ecosystem, tailored for building, scaling, and deploying large-scale recommendation systems,
            homepage_url: https://pytorch.org/torchrec/
            logo: torchrec.svg
            repo_url: https://github.com/pytorch/torchrec
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: torchgeo
            description: TorchGeo is a toolkit containing datasets, samplers, transforms, and pre-trained models for geospatial data.
            homepage_url: https://www.osgeo.org/projects/torchgeo/
            logo: TorchGeo.svg
            repo_url: https://github.com/microsoft/torchgeo
            crunchbase: https://www.crunchbase.com/organization/microsoft
      - subcategory:
        name: Medical & Biology
        items:
          - item:
            name: Clinica DL
            description: open-source deep learning software for reproducible neuroimaging processing. It can be seen as the deep learning extension of Clinica, an open-source Python library for neuroimaging preprocessing and analysis. 
            homepage_url: https://clinicadl.readthedocs.io/en/latest/
            logo: clinicaDL.svg
            repo_url: https://github.com/aramis-lab/clinicadl
            crunchbase: https://www.crunchbase.com/organization/sorbonne-university
          - item:
            name: FuseMedML
            description: A python framework accelerating ML based discovery in the medical field by encouraging code reuse
            homepage_url: https://pytorch.org/ecosystem/
            logo: FuseMedMLlogo.svg
            repo_url: https://github.com/BiomedSciAI/fuse-med-ml
            crunchbase: https://www.crunchbase.com/organization/ibm-research
          - item:
            name: GaNDLF
            description: The Generally Nuanced Deep Learning Framework (GaNDLF) for image segmentation and classification, especially in the biomedical field.
            homepage_url: https://www.nature.com/articles/s44172-023-00066-3
            logo: GaNDLF.svg
            repo_url: https://github.com/mlcommons/gandlf
            crunchbase: https://www.crunchbase.com/organization/mlcommons
          - item:
            name: MONAI
            description: AI Toolkit for Healthcare Imaging
            homepage_url: https://monai.io/
            logo: MONAI_Banner.svg
            repo_url: https://github.com/Project-MONAI
          - item:
            name: TIAToolbox
            description: TIAToolbox is a computational pathology toolbox for pathology image analysis.
            homepage_url: https://warwick.ac.uk/fac/cross_fac/tia/
            logo: TIAToolbox.svg
            repo_url: https://github.com/TissueImageAnalytics/tiatoolbox
          - item:
            name: torchdrug
            description: A powerful and flexible machine learning platform for drug discovery
            homepage_url: https://torchdrug.ai/
            logo: TorchDrug.svg
            repo_url: https://github.com/DeepGraphLearning/torchdrug
          - item:
            name: TorchIO
            description: Medical imaging toolkit for deep learning
            homepage_url: http://www.torchio.org/
            logo: TorchIO.svg
            repo_url: https://github.com/fepegar/torchio
      - subcategory:
        name: Adversarial & Robustness
        items:
          - item:
            name: AdverTorch
            description: A Toolbox for Adversarial Robustness Research
            homepage_url: https://advertorch.readthedocs.io/en/latest/
            logo: AdverTorch.svg
            repo_url: https://github.com/BorealisAI/advertorch
            crunchbase: https://www.crunchbase.com/organization/ecopia-tech-corporation
          - item:
            name: Captum
            description: An open source, extensible library for model interpretability built on PyTorch.
            homepage_url: https://captum.ai/docs/introduction.html
            logo: Captum.svg
            repo_url: https://github.com/pytorch/captum
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Ensemble-Pytorch
            description: unified ensemble framework for PyTorch to improve the performance and robustness of your ensemble based deep learning model
            homepage_url: https://ensemble-pytorch.readthedocs.io/
            logo: Ensemble_PyTorch.svg
            repo_url: https://github.com/TorchEnsemble-Community/Ensemble-Pytorch
          - item:
            name: pytorchfi
            description: A runtime fault injection tool for PyTorch
            homepage_url: https://pytorchfi.dev/
            logo: PyTorchFI.svg
            repo_url: https://github.com/pytorchfi/pytorchfi
          - item:
            name: TorchDrift
            description: TorchDrift is a data and concept drift library for PyTorch. It lets you monitor your PyTorch models to see if they operate within spec.
            homepage_url: https://torchdrift.org/
            logo: TorchDrift.svg
            repo_url: https://github.com/TorchDrift/TorchDrift
      - subcategory:
        name: Multimodal
        items:
          - item:
            name: MMF
            description: A modular framework for vision & language multimodal research.
            homepage_url: https://mmf.sh/
            logo: mmf.svg
            repo_url: https://github.com/facebookresearch/mmf
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: NeMo
            description: A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
            homepage_url: https://docs.nvidia.com/nemo-framework/user-guide/latest/overview.html
            logo: NEMO.svg
            repo_url: https://github.com/NVIDIA/NeMo
            crunchbase: https://www.crunchbase.com/organization/nvidia
          - item:
            name: USB
            description: Benchmark tool for developing and evaluating Semi-suprevised learning algorithms.  Includes an implementation of 14 SSL algorithms and 15 tasks for evaluation for CV, NLP, and Audio
            homepage_url: https://usb.readthedocs.io/en/main/
            logo: USB.svg
            repo_url: https://github.com/microsoft/Semi-supervised-learning
            crunchbase: https://www.crunchbase.com/organization/microsoft
      - subcategory:
        name: 3D
        items:
          - item:
            name: PyTorch-Points3d
            description: A framework for developing and testing common deep learning models to solve tasks related to unstructured 3D spatial data
            homepage_url: https://torch-points3d.readthedocs.io/en/latest/
            logo: PyTorch_Points3D.svg
            repo_url: https://github.com/torch-points3d/torch-points3d
          - item:
            name: PyTorch3D
            description: PyTorch3D is a library of reusable components for deep learning with 3D data
            homepage_url: https://pytorch3d.org/
            logo: PyTorch3D.svg
            repo_url: https://github.com/facebookresearch/pytorch3d
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: RoMa
            description: RoMa (which stands for Rotation Manipulation) provides differentiable mappings between 3D rotation representations, mappings from Euclidean to rotation space, and various utilities related to rotations.
            homepage_url: https://naver.github.io/roma/
            logo: RoMa.svg
            repo_url: https://github.com/naver/roma
      - subcategory:
        name: Quantum
        items:
          - item:
            name: PennyLane
            description: PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
            homepage_url: https://pennylane.ai/
            logo: PennyLane.svg
            repo_url: https://github.com/PennyLaneAI/pennylane
      - subcategory:
        name: Reinforcement Learning
        items:
          - item:
            name: torchrl
            description: TorchRL is an open-source Reinforcement Learning (RL) library for PyTorch.
            homepage_url: https://pytorch.org/rl/stable/index.html
            logo: torchrl.svg
            repo_url: https://github.com/pytorch/rl
            crunchbase: https://www.crunchbase.com/organization/facebook
  - category:
    name: Training
    subcategories:
      - subcategory:
        name: General
        items:
          - item:
            name: AdaptDL
            description: AdaptDL is a resource-adaptive deep learning (DL) training and scheduling framework, and is part of the CASL open source project.
            homepage_url: https://adaptdl.readthedocs.io/en/latest/
            logo: adaptdl.svg
            repo_url: https://github.com/petuum/adaptdl
            crunchbase: https://www.crunchbase.com/organization/petuum
          - item:
            name: Catalyst
            description: Catalyst is a PyTorch framework for Deep Learning Research and Development. It focuses on reproducibility, rapid experimentation, and codebase reuse so you can create something new rather than write yet another train loop.
            homepage_url: https://catalyst-team.com/
            logo: clearml-logo.svg
            repo_url: https://github.com/catalyst-team/catalyst
            crunchbase: https://www.crunchbase.com/organization/open-collective-75c2
          - item:
            name: ColossalAI
            description: Tools to kickstart distributed training and inference
            homepage_url: https://www.colossalai.org/
            logo: ColossalAI.svg
            repo_url: https://github.com/hpcaitech/ColossalAI
            crunchbase: https://www.crunchbase.com/organization/hpc-ai-technology
          - item:
            name: composer
            description: A open-source deep learning training library optimized for scalability and usability, integrating best practices for efficient, multi-node training
            homepage_url: http://docs.mosaicml.com/
            logo: composerMosaicML.svg
            repo_url: https://github.com/mosaicml/composer
            crunchbase: https://www.crunchbase.com/organization/databricks
          - item:
            name: fastai
            description: fastai is a deep learning library which provides practitioners with high-level components that can quickly and easily provide state-of-the-art results in standard deep learning domains, and provides researchers with low-level components that can be mixed and matched to build new approaches
            homepage_url: https://docs.fast.ai/
            logo: fastai.svg
            repo_url: https://github.com/fastai/fastai
            crunchbase: https://www.crunchbase.com/organization/fast-ai
          - item:
            name: Hummingbird
            description: Hummingbird is a library for compiling trained traditional ML models into tensor computations. Hummingbird allows users to seamlessly leverage neural network frameworks (such as PyTorch) to accelerate traditional ML models. 
            homepage_url: https://github.com/microsoft/hummingbird
            logo: hummingbird.svg
            repo_url: https://github.com/microsoft/hummingbird
            crunchbase: https://www.crunchbase.com/organization/microsoft
          - item:
            name: Ignite
            description: High-level library to help with training and evaluating neural networks in PyTorch
            homepage_url: https://pytorch-ignite.ai/
            logo: ignite_mixed.svg
            repo_url: https://github.com/pytorch/ignite
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: ludwig
            description: Ludwig is a low-code framework for building custom AI models like LLMs and other deep neural networks.
            homepage_url: https://ludwig.ai/latest/
            logo: ludwig_hero_smaller.svg
            repo_url: https://github.com/ludwig-ai/ludwig
          - item:
            name: Poutyne
            description: Poutyne is a simplified framework for PyTorch and handles much of the boilerplating code needed to train neural networks.
            homepage_url: https://poutyne.org/
            logo: Poutyne.svg
            repo_url: https://github.com/GRAAL-Research/poutyne
          - item:
            name: PyTorch Geometric Temporal
            description: Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
            homepage_url: https://lightning.ai/
            logo: lightning.svg
            repo_url: https://github.com/Lightning-AI/pytorch-lightning
            crunchbase: https://www.crunchbase.com/organization/grid-ai
          - item:
            name: skorch
            description: A scikit-learn compatible neural network library that wraps PyTorch.
            homepage_url: https://github.com/skorch-dev/skorch
            logo: skorch.svg
            repo_url: https://github.com/skorch-dev/skorch
          - item:
            name: stoke
            description: A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices (e.g. CPU, GPU), distributed modes, mixed-precision, and PyTorch extensions. This allows you to switch from local full-precision CPU to mixed-precision distributed multi-GPU with extensions (like optimizer state sharding) by simply changing a few declarative flags. 
            homepage_url: https://fidelity.github.io/stoke/
            logo: stoke.svg
            repo_url: https://github.com/StanfordPL/stoke
          - item:
            name: torchdistill
            description: Offers various state-of-the-art knowledge distillation methods and enables you to design (new) experiments simply by editing a declarative yaml config file.
            homepage_url: https://yoshitomo-matsubara.net/torchdistill/
            logo: torchdistill.svg
            repo_url: https://github.com/yoshitomo-matsubara/torchdistill
          - item:
            name: torchdata
            description: A Beta library of common modular data loading primitives for easily constructing flexible and performant data pipelines. 
            homepage_url: https://pytorch.org/data/beta/index.html
            logo: torchdata.svg
            repo_url: https://github.com/pytorch/data
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: TorchMetrics
            description: Machine learning metrics for distributed, scalable PyTorch applications.
            homepage_url: https://lightning.ai/docs/torchmetrics/
            logo: Torchmetrics.svg
            repo_url: https://github.com/Lightning-AI/torchmetrics
            crunchbase: https://www.crunchbase.com/organization/grid-ai
      - subcategory:
        name: Probabilistic & Optimization
        items:
          - item:
            name: baal
            description: Baal is a Bayesian active learning library. Provides methods to do posterior distribution sampling in order to maximize the efficiency of labelling during active learning. Our library is suitable for research and industrial applications.
            homepage_url: https://baal.readthedocs.io/
            logo: Baal.svg
            repo_url: https://github.com/baal-org/baal
            crunchbase: https://www.crunchbase.com/organization/element-ai
          - item:
            name: BoTorch
            description: A Framework for Efficient Monte-Carlo Bayesian Optimization
            homepage_url: https://botorch.org/
            logo: BoTorch.svg
            repo_url: https://github.com/pytorch/botorch
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: GPyTorch
            description: GPyTorch is a Gaussian process library implemented using PyTorch. GPyTorch is designed for creating scalable, flexible, and modular Gaussian process models with ease
            homepage_url: https://arxiv.org/abs/1809.11165
            logo: GPyTorch.svg
            repo_url: https://github.com/cornellius-gp/gpytorch
            crunchbase: https://www.crunchbase.com/organization/university-of-pennsylvania
          - item:
            name: Optuna
            description: Optuna is an automatic hyperparameter optimization software framework, particularly designed for machine learning.
            homepage_url: https://optuna.org/
            logo: optuna-logo.svg
            repo_url: https://github.com/optuna/optuna
          - item:
            name: Pomegranate
            description: pomegranate is a Python package that implements fast and flexible probabilistic models ranging from individual probability distributions to compositional models such as Bayesian networks and hidden Markov models
            homepage_url: https://pomegranate.readthedocs.io/en/latest/
            logo: pomegranate.svg
            repo_url: https://github.com/jmschrei/pomegranate
          - item:
            name: Pyro
            description: Deep universal probabilistic programming with Python and PyTorch
            homepage_url: https://pyro.ai/
            logo: Pyro.svg
            repo_url: https://github.com/pyro-ppl/pyro
      - subcategory:
        name: Privacy
        items:
          - item:
            name: CrypTen
            description: CrypTen is a Privacy Preserving Machine Learning framework written using PyTorch to train models using encrypted data. It is currently not production ready and its main use is as a research framework 
            homepage_url: https://crypten.readthedocs.io/en/latest/
            logo: CrypTen.svg
            repo_url: https://github.com/facebookresearch/CrypTen
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Opacus
            description: Train PyTorch models with Differential Privacy
            homepage_url: https://opacus.ai/
            logo: opacus_logo.svg
            repo_url: https://github.com/pytorch/opacus
          - item:
            name: PySyft
            description: Perform data science on data that remains in someone else's server
            homepage_url: https://www.openmined.org/
            logo: PySyft.svg
            repo_url: https://github.com/OpenMined/PySyft
      - subcategory:
        name: Continuous Learning
        items:
          - item:
            name: avalanche
            description: Avalanche is an End-to-End Continual Learning Library based on PyTorch, for fast prototyping, training, and reproducible evaluation of continual learning algorithms.
            homepage_url: http://avalanche.continualai.org/
            logo: Avalanche.svg
            repo_url: https://github.com/ContinualAI/avalanche
            crunchbase: https://www.crunchbase.com/organization/continualai
          - item:
            name: Renate
            description: Renate is a Python package for automatic retraining of neural networks models. It uses advanced Continual Learning and Lifelong Learning algorithms to achieve this purpose.
            homepage_url: https://renate.readthedocs.io/en/latest/
            logo: Renate.svg
            repo_url: https://github.com/awslabs/renate
            crunchbase: https://www.crunchbase.com/organization/amazon-web-services
      - subcategory:
        name: Federated Learning
        items:
          - item:
            name: Flower
            description: A unified approach to federated learning, analytics, and evaluation. Federate any workload, any ML framework, and any programming language
            homepage_url: https://flower.ai/
            logo: flower.svg
            repo_url: https://github.com/adap/flower
            crunchbase: https://www.crunchbase.com/organization/flower-1de8
          - item:
            name: Substra
            description: Substra is used to run complex federated learning experiments at scale.
            homepage_url: https://docs.substra.org/
            logo: Substra.svg
            repo_url: https://github.com/Substra
      - subcategory:
        name: Reinforcement Learning
        items:
          - item:
            name: PFRL
            description: PFRL is a PyTorch-based deep reinforcement learning library
            homepage_url: https://github.com/pfnet/pfrl
            logo: PFRL.svg
            repo_url: https://github.com/pfnet/pfrl
          - item:
            name: Stable Baselines3
            description: Stable Baselines3 (SB3) is a set of implementations of reinforcement learning algorithms
            homepage_url: https://stable-baselines3.readthedocs.io/
            logo: Stable_Baselines3.svg
            repo_url: https://github.com/DLR-RM/stable-baselines3
      - subcategory:
        name: Graph
        items:
          - item:
            name: PyTorch Geometric
            description: Graph Neural Network Library for PyTorch
            homepage_url: https://pyg.org/
            logo: PyTorch_Geometric.svg
            repo_url: https://github.com/pyg-team/pytorch_geometric
          - item:
            name: PyTorch Geometric Temporal
            description: PyTorch Geometric Temporal is a Spatiotemporal Signal Processing with Neural Machine Learning Models
            homepage_url: https://github.com/benedekrozemberczki/pytorch_geometric_temporal
            logo: PyTorch_Geometric_Temporal.svg
            repo_url: https://github.com/benedekrozemberczki/pytorch_geometric_temporal
      - subcategory:
        name: Multimodal
        items:
          - item:
            name: PyKale
            description: PyKale has a unified pipeline-based API and focuses on multimodal learning and transfer learning for graphs, images, and videos at the moment, with supporting models on deep learning and dimensionality reduction.
            homepage_url: https://pykale.github.io/
            logo: PyKale.svg
            repo_url: https://github.com/pykale/pykale
      - subcategory:
        name: Self supervised
        items:
          - item:
            name: PyTorch Metric Learning
            description: Modular, flexible, and extensible library for deep metric learning.
            homepage_url: https://kevinmusgrave.github.io/pytorch-metric-learning/
            logo: PyTorch_Metric_Learning.svg
            repo_url: https://github.com/KevinMusgrave/pytorch-metric-learning
      - subcategory:
        name: Distributed
        items:
          - item:
            name: Ray
            description: Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
            homepage_url: https://github.com/ray-project/ray
            logo: ray.svg
            repo_url: https://github.com/ray-project/ray
      - subcategory:
        name: Quantum
        items:
          - item:
            name: TorchQuantum
            description: A framework for Quantum Classical Simulation, Quantum Machine Learning, Quantum Neural Networks, Parameterized Quantum Circuits with support for easy deployments on real quantum computers.
            homepage_url: https://hanruiwanghw.wixsite.com/torchquantum
            logo: Torch_Quantum.svg
            repo_url: https://github.com/mit-han-lab/torchquantum
  - category:
    name: Optimizations
    subcategories:
      - subcategory:
        name: General
        items:
          - item:
            name: einops
            description: Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
            homepage_url: https://einops.rocks/
            logo: einops.svg
            repo_url: https://github.com/arogozhnikov/einops
            crunchbase: https://www.crunchbase.com/organization/chai-discovery
          - item:
            name: higher
            description: higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps. 
            homepage_url: https://higher.readthedocs.io/en/latest/.
            logo: higher.svg
            repo_url: https://github.com/facebookresearch/higher
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: ivy
            description: Convert Machine Learning Code Between Frameworks
            homepage_url: https://ivy.dev/
            logo: ivylogo.svg
            repo_url: https://github.com/ivy-llc/ivy
            crunchbase: https://www.crunchbase.com/organization/ivy
          - item:
            name: neural-compressor
            description: An open-source Python library supporting popular model compression techniques on all deep learning frameworks.
            homepage_url: https://intel.github.io/neural-compressor/latest/docs/source/Welcome.html
            logo: IntelNeuralCompressor.svg
            repo_url: https://github.com/intel/neural-compressor
            crunchbase: https://www.crunchbase.com/organization/intel
          - item:
            name: TensorLy
            description: TensorLy is a Python library that aims at making tensor learning simple and accessible. It allows to easily perform tensor decomposition, tensor learning and tensor algebra. Its backend system allows to seamlessly perform computation with NumPy, PyTorch, JAX, TensorFlow, CuPy or Paddle, and run methods at scale on CPU or GPU.
            homepage_url: http://tensorly.org/
            logo: tensorly.svg
            repo_url: https://github.com/tensorly/tensorly
          - item:
            name: torcharrow
            description: TorchArrow is a torch.Tensor-like Python DataFrame library for data preprocessing in deep learning. It supports multiple execution runtimes and Arrow as a common format.
            homepage_url: https://pytorch.org/torcharrow/beta/index.html
            logo: torcharrow.svg
            repo_url: https://github.com/pytorch/torcharrow
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: tensordict
            description: TensorDict is a dictionary-like class that inherits properties from tensors, such as indexing, shape operations, casting to device etc.
            homepage_url: https://pytorch.org/tensordict/stable/index.html
            logo: tensordict.svg
            repo_url: https://github.com/pytorch/tensordict
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: torchao
            description: PyTorch library for custom data types & optimizations. Quantize and sparsify weights, gradients, optimizers & activations for inference and training.
            homepage_url: https://pytorch.org/blog/pytorch-native-architecture-optimization/
            logo: torchao.svg
            repo_url: https://github.com/pytorch/ao
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Torchopt
            description: An efficient library for differentiable optimization built upon PyTorch
            homepage_url: https://torchopt.readthedocs.io/en/latest/
            logo: TorchOpt.svg
            repo_url: https://github.com/metaopt/TorchOpt
          - item:
            name: Zeus
            description: Zeus is a library for measuring the energy consumption of Deep Learning workloads and optimizing their energy consumption.
            homepage_url: https://ml.energy/zeus/
            logo: Zeus.svg
            repo_url: https://github.com/ml-energy/zeus
      - subcategory:
        name: Compilers & Runtimes
        items:
          - item:
            name: Depyf
            description: depyf is a tool to help you understand debug and get insights into pytorch.compile
            homepage_url: https://depyf.readthedocs.io/en/latest/
            logo: depfy.svg
            repo_url: https://github.com/thuml/depyf
          - item:
            name: Glow
            description: Glow is a machine learning compiler and execution engine for hardware accelerators. It is designed to be used as a backend for high-level machine learning frameworks. The compiler is designed to allow state of the art compiler optimizations and code generation of neural network graphs
            homepage_url: https://github.com/pytorch/glow
            logo: glow.svg
            repo_url: https://github.com/pytorch/glow
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: intel-extension-for-pytorch
            description: A Python extension to optimize performance on an Intel platform.
            homepage_url: https://intel.github.io/intel-extension-for-pytorch/
            logo: intelex.svg
            repo_url: https://github.com/intel/intel-extension-for-pytorch
            crunchbase: https://www.crunchbase.com/organization/intel
          - item:
            name: OctoML Profiler
            description: octoml-profile is a python library and cloud service that enables ML engineers to easily assess the performance and cost of PyTorch models on cloud hardware with state-of-the-art ML acceleration technology.
            homepage_url: https://github.com/octoml/octoml-profile
            logo: octoml.svg
            repo_url: https://github.com/octoml/octoml-profile
          - item:
            name: ONNX runtime
            description: High performance ML inferencing and training accelerator
            homepage_url: https://onnxruntime.ai/
            logo: ONNX_Runtime.svg
            repo_url: https://github.com/microsoft/onnxruntime
            crunchbase: https://www.crunchbase.com/organization/microsoft
          - item:
            name: PopTorch
            description: PopTorch is a set of extensions for PyTorch enabling models to be trained, evaluated and used on the Graphcore IPU.
            homepage_url: https://docs.graphcore.ai/projects/poptorch-user-guide/en/latest/
            logo: poptorch.svg
            repo_url: https://github.com/graphcore/poptorch
            crunchbase: https://www.crunchbase.com/organization/graphcore
          - item:
            name: Speedster
            description: Speedster reduces inference costs by leveraging SOTA optimization techniques that best couple your AI models with the underlying hardware (GPUs and CPUs).
            homepage_url: https://www.nebuly.com/
            logo: speedster.svg
            repo_url: https://github.com/nebuly-ai/optimate/tree/main/optimization/speedster
          - item:
            name: Torch-TensorRT
            description: Torch-TensorRT is a inference compiler for PyTorch, targeting NVIDIA GPUs via NVIDIA’s TensorRT Deep Learning Optimizer and Runtime.
            homepage_url: https://pytorch.org/TensorRT/
            logo: TensorRT.svg
            repo_url: https://github.com/RizhaoCai/PyTorch_ONNX_TensorRT
          - item:
            name: PyTorchXLA
            description: PyTorch runs on XLA devices, like TPUs, with the torch_xla package. 
            homepage_url: https://pytorch.org/xla/release/2.1/index.html
            logo: pytorchxla.svg
            repo_url: https://github.com/pytorch/xla
            crunchbase: https://www.crunchbase.com/organization/facebook
      - subcategory:
        name: MLOps
        items:
          - item:
            name: accelerate
            description: Accelerate is a library that enables the same PyTorch code to be run across distributed configurations
            homepage_url: https://huggingface.co/docs/accelerate/index
            logo: accelerate.svg
            repo_url: https://github.com/huggingface/accelerate
            crunchbase: https://www.crunchbase.com/organization/hugging-face
          - item:
            name: Clear ML
            description: Suite of tools to streamline your AI workflow.
            homepage_url: https://clear.ml/
            logo: clearml-logo.svg
            repo_url: https://github.com/allegroai/clearml
            crunchbase: https://www.crunchbase.com/organization/allegro-ai
          - item:
            name: Determined
            description: Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.
            homepage_url: https://determined.ai/
            logo: determined-logo.svg
            repo_url: https://github.com/determined-ai/determined
            crunchbase: https://www.crunchbase.com/organization/determined-ai
          - item:
            name: forte
            description: Forte is a flexible and powerful ML workflow builder. 
            homepage_url: https://github.com/asyml/forte
            logo: forte.svg
            repo_url: https://github.com/asyml/forte
            crunchbase: https://www.crunchbase.com/organization/petuum
          - item:
            name: Hydra
            description: simplifies the development of research and other complex applications. The key feature is the ability to dynamically create a hierarchical configuration by composition and override it through config files and the command line. The name Hydra comes from its ability to run multiple similar jobs - much like a Hydra with multiple heads.
            homepage_url: https://hydra.cc/
            logo: hydra.svg
            repo_url: https://github.com/facebookresearch/hydra
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: padl
            description: A pipeline builder for PyTorch.
            homepage_url: https://lf1-io.github.io/padl/latest/
            logo: padl.svg
            repo_url: https://github.com/lf1-io/padl
          - item:
            name: Polyaxon
            description: MLOps Tools For Managing & Orchestrating The Machine Learning LifeCycle
            homepage_url: https://polyaxon.com/
            logo: Polyaxon.svg
            repo_url: https://github.com/polyaxon/polyaxon
      - subcategory:
        name: Distributed
        items:
          - item:
            name: DeepSpeed
            description: DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
            homepage_url: https://www.deepspeed.ai/
            logo: DeepSpeed_light.svg
            repo_url: https://github.com/microsoft/DeepSpeed
            crunchbase: https://www.crunchbase.com/organization/hugging-face
          - item:
            name: FairScale
            description: PyTorch extension library for high performance and large scale training. This library extends basic PyTorch capabilities while adding new SOTA scaling techniques. 
            homepage_url: https://github.com/facebookresearch/fairscale
            logo: fairscale.svg
            repo_url: https://github.com/facebookresearch/fairscale
            crunchbase: https://www.crunchbase.com/organization/facebook
          - item:
            name: Horovod
            description: Horovod is a distributed deep learning training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
            homepage_url: http://horovod.ai/
            logo: HOROVOD.svg
            repo_url: https://github.com/horovod/horovod
            crunchbase: https://www.crunchbase.com/organization/lf-artificial-intelligence-foundation