TensorFlow-based neural network library
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
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 refreshing functional take on deep learning
A TensorFlow library for applied reinforcement learning
mlpack: a scalable C++ machine learning library
High-performance library for gradient boosting on decision trees
A fast image processing library with low memory needs
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
C++ DataFrame for statistical, Financial, and ML analysis
Detecting silent model failure. NannyML estimates performance
Python package built to ease deep learning on graph
A GPU-accelerated library containing highly optimized building blocks