Automate browser based workflows with AI

The absolute trainer to light up AI agents.

Animation engine for explanatory math videos

real time face swap and one-click video deepfake with only a single image

Simple SDF mesh generation in Python

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

Photometric optimization code for creating the FLAME texture space and other applications

The official code for "Efficient Neural Radiance Fields" in CVPR2022.

A large-scale face dataset for face parsing, recognition, generation and editing.

NeRF visualization library under construction

Official code of Training and Tuning Generative Neural Radiance Fields for Attribute-Conditional 3D-Aware Face Generation

An Efficient Implementation of Analytic Mesh Algorithm for 3D Iso-surface Extraction from Neural Networks

PyTorch implementations of Generative Adversarial Networks.

ViTOL

DECA: Detailed Expression Capture and Animation (SIGGRAPH 2021)

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

FFCV: Fast Forward Computer Vision (and other ML workloads!)

A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.

3d plotting for Python in the Jupyter notebook based on IPython widgets using WebGL

Perception toolkit for sim2real training and validation in Unity

Toolbox of models, callbacks, and datasets for AI/ML researchers.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Toolkit to run Python benchmarks

Pythonic task management & command execution.

A Python package that collects robotic environments based on the PyBullet simulator, suitable to develop and test Reinforcement Learning algorithms on simulated grasping and manipulation applications.

Robotics Toolbox for Python

pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, DeepSeek, Mixtral, Gemma, Phi, MiniCPM, Qwen-VL, MiniCPM-V, etc.) on Intel XPU (e.g., local PC with iGPU and NPU, discr…