Petastorm library enables single machine or distributed training
Open Source Differentiable Computer Vision Library
Implementation of Imagen, Google's Text-to-Image Neural Network
Training PyTorch models with differential privacy
Open source machine learning framework to automate text conversations
A python library for self-supervised learning on images
machine learning tutorials (mainly in Python3)
Fast image augmentation library and an easy-to-use wrapper
Python library for portfolio optimization built on top of scikit-learn
A library for scientific machine learning & physics-informed learning
Deep learning library
A python library for easy manipulation and forecasting of time series
DoWhy is a Python library for causal inference
PyTorch version of Stable Baselines
Core ML tools contain supporting tools for Core ML model conversion
A unified, comprehensive and efficient recommendation library
Tool for visualizing and tracking your machine learning experiments
Library for OCR-related tasks powered by Deep Learning
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Physical Symbolic Optimization
Geometric deep learning extension library for PyTorch
Training data (data labeling, annotation, workflow) for all data types
End-to-End Library for Continual Learning based on PyTorch
Making large AI models cheaper, faster and more accessible