TensorFlow-based neural network library
Build a machine learning model from a prompt
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
A library for accelerating Transformer models on NVIDIA GPUs
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Library for OCR-related tasks powered by Deep Learning
Fast, flexible and easy to use probabilistic modelling in Python
A TensorFlow library for applied reinforcement learning
MiniSom is a minimalistic implementation of the Self Organizing Maps
DeepMind's software stack for physics-based simulation
A comprehensive set of fairness metrics for datasets
A lightweight vision library for performing large object detection
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
A refreshing functional take on deep learning
Detecting silent model failure. NannyML estimates performance
Python package built to ease deep learning on graph
Hub of ready-to-use datasets for ML models
AIMET is a library that provides advanced quantization and compression
Train machine learning models within Docker containers
Scalable machine learning for time series forecasting
A library for deep learning end-to-end dialog systems and chatbots
Scalable and user friendly neural forecasting algorithms.
The Unified Machine Learning Framework