Differentiable SDE solvers with GPU support and efficient sensitivity
Detecting silent model failure. NannyML estimates performance
Decentralized deep learning in PyTorch. Built to train models
Interpretability and explainability of data and machine learning model
PyTorch extensions for fast R&D prototyping and Kaggle farming
OpenMMLab Image Classification Toolbox and Benchmark
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
Spatiotemporal Signal Processing with Neural Machine Learning Models
A comprehensive set of fairness metrics for datasets
A unified, comprehensive and efficient recommendation library
Probabilistic reasoning and statistical analysis in TensorFlow
Data science on data without acquiring a copy
Embed images and sentences into fixed-length vectors
Python package built to ease deep learning on graph
Geometric deep learning extension library for PyTorch
A lightweight 3D Morphable Face Model library in modern C++
The unified and scalable ML library for large-scale training
A system for quickly generating training data with weak supervision
Uniform Manifold Approximation and Projection
PyTorch version of Stable Baselines
Machine Learning Pipelines for Kubeflow
Your open-source LLM evaluation toolkit
NVIDIA Federated Learning Application Runtime Environment
A machine learning library for detecting anomalies in signals
Django friendly finite state machine support