PyTorch extensions for fast R&D prototyping and Kaggle farming
Probabilistic reasoning and statistical analysis in TensorFlow
Medical imaging toolkit for deep learning
DeepMind's software stack for physics-based simulation
A TensorFlow library for applied reinforcement learning
MiniSom is a minimalistic implementation of the Self Organizing Maps
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Library to help with training and evaluating neural networks
A library for accelerating Transformer models on NVIDIA GPUs
Open source platform for the machine learning lifecycle
Detecting silent model failure. NannyML estimates performance
Training PyTorch models with differential privacy
A comprehensive set of fairness metrics for datasets
A lightweight vision library for performing large object detection
Graph Neural Network Library for PyTorch
Serve machine learning models within a Docker container
Deep learning optimization library making distributed training easy
Geometric deep learning extension library for PyTorch
Uncover insights, surface problems, monitor, and fine tune your LLM
Implementation of 'lightweight' GAN, proposed in ICLR 2021
A refreshing functional take on deep learning
The Unified Machine Learning Framework
The easiest way to use deep metric learning in your application
A PyTorch-based Speech Toolkit
Hub of ready-to-use datasets for ML models