Aprendizaje automático · GitHub

Pool-based active learning in Python

ALiPy: Active Learning in Python is an active learning python toolbox, which allows users to conveniently evaluate, compare and analyze the performance of active learning methods.

A simple implementation of LeakGAN in PyTorch

We introduce a novel approach for parameter generation, named neural network parameter diffusion (p-diff), which employs a standard latent diffusion model to synthesize a new set of parameters

TensorFlow models accelerated with NVIDIA TensorRT

Deep learned, NVIDIA-accelerated 3D object pose estimation

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.

NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process communication and coordination overheads by allowing programmers to perform one-sided communication from within CUDA kernels and on CUDA streams.

A library that uses hardware acceleration to load sequences of video frames to facilitate machine learning training

A GPU performance profiling tool for PyTorch models

Fast and accurate object detection with end-to-end GPU optimization

Deep Learning Experiment Management

Official PyTorch implementation of Dreaming to Distill: Data-free Knowledge Transfer via DeepInversion (CVPR 2020)

EDM2 and Autoguidance -- Official PyTorch implementation

PyTorch Implementation of EmerNeRF: Emergent Spatial-Temporal Scene Decomposition via Self-Supervision

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

Official PyTorch implementation of FB-BEV & FB-OCC - Forward-backward view transformation for vision-centric autonomous driving perception

[ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule

OmniVinci is an omni-modal LLM for joint understanding of vision, audio, and language.

SPLATNet: Sparse Lattice Networks for Point Cloud Processing (CVPR2018)

ToolOrchestra is an end-to-end RL training framework for orchestrating tools and agentic workflows.