Lightning fast C++/CUDA neural network framework
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
A framework for real-life data science
A python library for self-supervised learning on images
Jittor is a high-performance deep learning framework
machine learning tutorials (mainly in Python3)
Fast image augmentation library and an easy-to-use wrapper
A fast image processing library with low memory needs
A library for scientific machine learning & physics-informed learning
Medical imaging toolkit for deep learning
A unified framework for scalable computing
Deep learning library
An Easy-to-use, Scalable and High-performance RLHF Framework
A python library for easy manipulation and forecasting of time series
DoWhy is a Python library for causal inference
Core ML tools contain supporting tools for Core ML model conversion
A unified, comprehensive and efficient recommendation library
The open-source tool for building high-quality datasets
Implementation of Imagen, Google's Text-to-Image Neural Network
AIMET is a library that provides advanced quantization and compression
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Python library for portfolio optimization built on top of scikit-learn
DeepMind's software stack for physics-based simulation
ML engineer that reads papers, trains models, and ships ML models