A library for scientific machine learning & physics-informed learning
Deep learning optimization library making distributed training easy
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
OpenVINO™ Toolkit repository
High-performance library for gradient boosting on decision trees
Machine Learning Toolkit for Kubernetes
A library for deep learning end-to-end dialog systems and chatbots
Multilingual Automatic Speech Recognition with word-level timestamps
Time series forecasting with PyTorch
Transfer learning / domain adaptation / domain generalization
Gradient boosting framework based on decision tree algorithms
Python library for defining and optimizing mathematical expressions
Implementation of DeepLabCut
Superfast AI decision making and processing of multi-modal data
Python package built to ease deep learning on graph
Serve, optimize and scale PyTorch models in production
Toolkit for conversational AI
YOLOv5 is the world's most loved vision AI
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
NVIDIA Federated Learning Application Runtime Environment
Graph Neural Network Library for PyTorch
Solve end to end problems using Llama model family
An AI for Music Generation
Lightning fast C++/CUDA neural network framework
A python library for self-supervised learning on images