Play couplet with seq2seq model
Running large language models on a single GPU
Transfer learning / domain adaptation / domain generalization
Machine Learning Pipelines for Kubeflow
Training PyTorch models with differential privacy
Easy-to-use,Modular and Extendible package of deep-learning models
JAX-based neural network library
Probabilistic reasoning and statistical analysis in TensorFlow
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
AI agents autonomously run and improve ML experiments overnight
Python examples of popular machine learning algorithms
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Petastorm library enables single machine or distributed training
A cross-platform Python library for differentiable programming
Spatiotemporal Signal Processing with Neural Machine Learning Models
A lightweight vision library for performing large object detection
Probabilistic time series modeling in Python
Open source framework for deep learning satellite and aerial imagery
Fast image augmentation library and an easy-to-use wrapper