A very simple framework for state-of-the-art NLP
TensorFlow-based neural network library
PyTorch extensions for fast R&D prototyping and Kaggle farming
A refreshing functional take on deep learning
AIMET is a library that provides advanced quantization and compression
Fast, flexible and easy to use probabilistic modelling in Python
Medical imaging toolkit for deep learning
A TensorFlow library for applied reinforcement learning
High-performance library for gradient boosting on decision trees
A library for accelerating Transformer models on NVIDIA GPUs
A lightweight 3D Morphable Face Model library in modern C++
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
DeepMind's software stack for physics-based simulation
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
C++ DataFrame for statistical, Financial, and ML analysis
A GPU-accelerated library containing highly optimized building blocks
The Operator Splitting QP Solver
Hub of ready-to-use datasets for ML models