High performance distributed framework for training deep learning recommendation models based on PyTorch.

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.

PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

Cockpit: A Practical Debugging Tool for Training Deep Neural Networks

Visual analysis and diagnostic tools to facilitate machine learning model selection.

Domain adaptation made easy. Fully featured, modular, and customizable.

Containers for machine learning

Python library for audio and music analysis

A deep learning library for video understanding research.

Visualizer for PyTorch image models

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Lucid library adapted for PyTorch

Kats, a kit to analyze time series data, a lightweight, easy-to-use, generalizable, and extendable framework to perform time series analysis, from understanding the key statistics and characteristi…

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

HiPlot makes understanding high dimensional data easy

Refactoring dalle-pytorch and taming-transformers for TPU VM

The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).

MADGRAD Optimization Method

Open Source Photos Platform Powered by PyTorch

Differentiable architecture search for convolutional and recurrent networks

Deep probabilistic analysis of single-cell and spatial omics data

OpenMMLab Multimodal Advanced, Generative, and Intelligent Creation Toolbox. Unlock the magic 🪄: Generative-AI (AIGC), easy-to-use APIs, awsome model zoo, diffusion models, for text-to-image genera…

Bayesian active learning library for research and industrial usecases.

Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)

The Research Tree - A playground for research at the intersection of Continual, Reinforcement, and Self-Supervised Learning.

Integrate pytorch

A treasure chest for visual classification and recognition powered by PaddlePaddle