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