org)

[

  * Substra 
  * ![](/assets/images/logo-github.svg)
  *   * 
Substra is a federated learning Python library to run federated learning
experiments at scale on real distributed data.  ](https://github.com/Substra)

[

  * Colossal-LLaMA-2 
  * ![](/assets/images/logo-github.svg)
  *   * 
A complete and open-sourced solution for injecting domain-specific knowledge
into pre-trained LLM.
](https://github.com/hpcaitech/ColossalAI/tree/main/applications/Colossal-
LLaMA-2)

[

  * PyTorchVideo 
  * ![](/assets/images/logo-github.svg)
  *   * 
A deep learning library for video understanding research. Hosts various video-
focused models, datasets, training pipelines and more.
](https://pytorchvideo.org/)

[

  * neural-compressor 
  * ![](/assets/images/logo-github.svg)
  *   * 
Intel® Neural Compressor provides unified APIs for network compression
technologies for faster inference  ](https://intel.github.io/neural-
compressor/)

[

  * Poutyne 
  * ![](/assets/images/logo-github.svg)
  *   * 
Poutyne is a Keras-like framework for PyTorch and handles much of the
boilerplating code needed to train neural networks.  ](https://poutyne.org/)

[

  * DeepSpeed 
  * ![](/assets/images/logo-github.svg)
  *   * 
DeepSpeed is a deep learning optimization library that makes distributed
training easy, efficient, and effective.  ](https://www.deepspeed.ai/)

[

  * accelerate 
  * ![](/assets/images/logo-github.svg)
  *   * 
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-
precision  ](https://huggingface.co/docs/accelerate)

[

  * OpenMMLab 
  * ![](/assets/images/logo-github.svg)
  *   * 
OpenMMLab covers a wide range of computer vision research topics including
classification, detection, segmentation, and super-resolution.
](https://github.com/open-mmlab)

[

  * TensorLy 
  * ![](/assets/images/logo-github.svg)
  *   * 
TensorLy is a high level API for tensor methods and deep tensorized neural
networks in Python that aims to make tensor learning simple.
](http://tensorly.org/stable/home.html)

[

  * Flair 
  * ![](/assets/images/logo-github.svg)
  *   * 
Flair is a very simple framework for state-of-the-art natural language
processing (NLP).  ](https://github.com/flairNLP/flair)

[

  * RoMa 
  * ![](/assets/images/logo-github.svg)
  *   * 
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.  ](https://github.com/naver/roma)

[

  * joeynmt 
  * ![](/assets/images/logo-github.svg)
  *   * 
Minimalist Neural Machine Translation toolkit for educational purposes
](https://joeynmt.readthedocs.io/en/latest/)

[

  * stoke 
  * ![](/assets/images/logo-github.svg)
  *   * 
A lightweight declarative PyTorch wrapper for context switching between
devices, distributed modes, mixed-precision, and PyTorch extensions.
](https://fidelity.github.io/stoke/)

[

  * Diffusers 
  * ![](/assets/images/logo-github.svg)
  *   * 
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.  ](https://huggingface.co/docs/diffusers)

[

  * Horovod 
  * ![](/assets/images/logo-github.svg)
  *   * 
Horovod is a distributed training library for deep learning frameworks.
Horovod aims to make distributed DL fast and easy to use.
](http://horovod.ai)

[

  * PyTorch3D 
  * ![](/assets/images/logo-github.svg)
  *   * 
PyTorch3D provides efficient, reusable components for 3D Computer Vision
research with PyTorch.  ](https://pytorch3d.org/)

[

  * Stable Baselines3 
  * ![](/assets/images/logo-github.svg)
  *   * 
Stable Baselines3 (SB3) is a set of reliable implementations of reinforcement
learning algorithms in PyTorch.  ](https://github.com/DLR-RM/stable-
baselines3)

[

  * Polyaxon 
  * ![](/assets/images/logo-github.svg)
  *   * 
Polyaxon is a platform for building, training, and mo