A library for accelerating Transformer models on NVIDIA GPUs
Scalable and user friendly neural forecasting algorithms.
Machine Learning Pipelines for Kubeflow
Trainable models and NN optimization tools
End-to-End Library for Continual Learning based on PyTorch
Decentralized deep learning in PyTorch. Built to train models
Training PyTorch models with differential privacy
Medical imaging toolkit for deep learning
Test Suites for validating ML models & data
The open-source tool for building high-quality datasets
Easy-to-use,Modular and Extendible package of deep-learning models
JAX-based neural network library
Build portable, production-ready MLOps pipelines
Time series Timeseries Deep Learning Machine Learning Pytorch fastai
DeepVariant is an analysis pipeline that uses a deep neural networks
Implementation of DeepLabCut
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
The easiest way to use deep metric learning in your application
Geometric deep learning extension library for PyTorch
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