Modular quant framework
Library to help with training and evaluating neural networks
Making Enterprise Data Intelligent and Responsive for AI
A library for accelerating Transformer models on NVIDIA GPUs
Scalable and user friendly neural forecasting algorithms.
Your open-source LLM evaluation toolkit
End-to-End Library for Continual Learning based on PyTorch
Training PyTorch models with differential privacy
Test Suites for validating ML models & data
The open-source tool for building high-quality datasets
High-level training, data augmentation, and utilities for Pytorch
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
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
A machine learning library for detecting anomalies in signals
Proofs, cases, concept supplements, and reference explanations
A library for scientific machine learning & physics-informed learning
A lightweight vision library for performing large object detection
AutoGluon: AutoML for Image, Text, and Tabular Data
Create HTML profiling reports from pandas DataFrame objects
Train machine learning models within Docker containers
Django friendly finite state machine support
Implementation of Imagen, Google's Text-to-Image Neural Network