JAX-based neural network library
Hummingbird compiles trained ML models into tensor computation
Embed images and sentences into fixed-length vectors
Haystack is an open source NLP framework to interact with your data
Implementation of DeepLabCut
Fast image augmentation library and an easy-to-use wrapper
A library for scientific machine learning & physics-informed learning
Implementation of Imagen, Google's Text-to-Image Neural Network
Graph Neural Network Library for PyTorch
48khz stereo neural audio codec for general audio
Python library for Representation Learning on Knowledge Graphs
Source code of PyGAD, Python 3 library for building genetic algorithms
Open-source choice to scale, assess and maintain natural language data
OpenMMLab Model Deployment Framework
Training PyTorch models with differential privacy
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Time series forecasting with PyTorch
A python library for easy manipulation and forecasting of time series
Libraries for applying sparsification recipes to neural networks
ktrain is a Python library that makes deep learning AI more accessible
A Python package for extending the official PyTorch
Python toolbox to create adversarial examples
JAX-based neural network library
Core ML tools contain supporting tools for Core ML model conversion