PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
Medical imaging toolkit for deep learning
Fast, flexible and easy to use probabilistic modelling in Python
AIMET is a library that provides advanced quantization and compression
A TensorFlow library for applied reinforcement learning
A refreshing functional take on deep learning
A library for accelerating Transformer models on NVIDIA GPUs
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
DeepMind's software stack for physics-based simulation
Library for OCR-related tasks powered by Deep Learning
A comprehensive set of fairness metrics for datasets
Geometric deep learning extension library for PyTorch
A lightweight vision library for performing large object detection
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
A PyTorch-based Speech Toolkit
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
Scalable and user friendly neural forecasting algorithms.
The Unified Machine Learning Framework
The easiest way to use deep metric learning in your application
Models and examples built with TensorFlow
Train machine learning models within Docker containers