Running large language models on a single GPU
Minimal and clean examples of machine learning algorithms
Transfer learning / domain adaptation / domain generalization
Learn how to develop, deploy and iterate on production-grade ML
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
An Easy-to-use, Scalable and High-performance RLHF Framework
Standalone, small, language-neutral
TFX is an end-to-end platform for deploying production ML pipelines
TimeGPT-1: production ready pre-trained Time Series Foundation Model
Pretrained (Language) Models for Probabilistic Time Series Forecasting
Python framework for adversarial attacks, and data augmentation
A reactive notebook for Python
We write your reusable computer vision tools
MLOps simplified. From ML Pipeline ⇨ Data Product without the hassle
End-to-End Library for Continual Learning based on PyTorch
Training PyTorch models with differential privacy
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Medical imaging toolkit for deep learning
Simulation of spiking neural networks (SNNs) using PyTorch
Adversarial Robustness Toolbox (ART) - Python Library for ML security
A Python package for extending the official PyTorch
An unsupervised and free tool for image and video dataset analysis
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
NVIDIA Federated Learning Application Runtime Environment